Commit ed8b8e9
authored
Support negative dimensions on "aten.split_with_sizes_copy.default" (#16006)
### Summary
Support a negative dimension for MPS SplitWithSizes implementation.
### Test plan
I re-registered this node visitor on a model that previously failed to
export with the error message
```
RuntimeError: split_copy: dim -1 out of range for input tensor with 2 dimensions
```
and it succeeded.1 parent 488d761 commit ed8b8e9
1 file changed
+4
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
242 | 242 | | |
243 | 243 | | |
244 | 244 | | |
| 245 | + | |
245 | 246 | | |
| 247 | + | |
| 248 | + | |
246 | 249 | | |
247 | 250 | | |
248 | 251 | | |
249 | | - | |
| 252 | + | |
250 | 253 | | |
251 | 254 | | |
252 | 255 | | |
| |||
0 commit comments