Repository navigation
Commit a2ae4b3
authored
Qualcomm AI Engine Direct - Preserve partition metadata on single-output delegate getitem nodes (pytorch#20805)
### Summary
The single-output path in `create_submodule_from_nodes` copied only
"val" onto the manually-built getitem, dropping other partition meta
(e.g. QNN's `q_tensor_io` tag) that the multi-output path keeps via
`propagate_meta`, leaving backend IO-tagging passes with wrong delegate
`dtype`/`spec`.
Fix by propagating full meta then dropping
`nn_module_stack`/`source_fn_stack`, matching the multi-output path.1 parent 7a6f630 commit a2ae4b3
1 file changed
Lines changed: 3 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
884 | 884 | | |
885 | 885 | | |
886 | 886 | | |
887 | | - | |
888 | | - | |
| 887 | + | |
889 | 888 | | |
890 | 889 | | |
| 890 | + | |
| 891 | + | |
891 | 892 | | |
892 | 893 | | |
893 | 894 | | |
| |||
0 commit comments