Commit b57875b
committed
Fix aten_unbind for torch >= 2.7 dynamo export
Replace Split op with explicit Slice operations to fix TypeError when
unbind is called during ONNX export with dynamo=True. The Split op
with num_outputs parameter returns a non-iterable SymbolicTensor
instead of a sequence, causing the list comprehension to fail.
The fix uses individual Slice + Squeeze operations for each output,
which properly handles symbolic tensors during graph construction.
Fixes pytorch/pytorch#1689691 parent 9dbf685 commit b57875b
1 file changed
+15
-8
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
9201 | 9201 | | |
9202 | 9202 | | |
9203 | 9203 | | |
9204 | | - | |
9205 | | - | |
| 9204 | + | |
| 9205 | + | |
9206 | 9206 | | |
9207 | | - | |
9208 | | - | |
9209 | | - | |
9210 | | - | |
9211 | | - | |
9212 | | - | |
| 9207 | + | |
| 9208 | + | |
| 9209 | + | |
| 9210 | + | |
| 9211 | + | |
| 9212 | + | |
| 9213 | + | |
| 9214 | + | |
| 9215 | + | |
| 9216 | + | |
| 9217 | + | |
| 9218 | + | |
| 9219 | + | |
9213 | 9220 | | |
9214 | 9221 | | |
9215 | 9222 | | |
| |||
0 commit comments