Commit 459d383
[ET-VK][ez] Fix IndexError in Vulkan partitioner DtypeSetList/TensorRepSetList
Pull Request resolved: #18048
The `__getitem__` methods of `DtypeSetList` and `TensorRepSetList` in
`utils.py` could raise an `IndexError` when the index is greater than or
equal to the length of the list. This can happen when partitioning ops
whose number of inputs or outputs exceeds the number of entries in the
dtype/tensor-rep specification list. Fix by returning an empty set in
this case, matching the intent of the existing broadcasting logic.
ghstack-source-id: 353546684
@exported-using-ghexport
Differential Revision: [D95970163](https://our.internmc.facebook.com/intern/diff/D95970163/)1 parent 0eb0585 commit 459d383
1 file changed
Lines changed: 5 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
91 | 91 | | |
92 | 92 | | |
93 | 93 | | |
| 94 | + | |
| 95 | + | |
94 | 96 | | |
95 | 97 | | |
96 | 98 | | |
| |||
1227 | 1229 | | |
1228 | 1230 | | |
1229 | 1231 | | |
1230 | | - | |
1231 | | - | |
| 1232 | + | |
| 1233 | + | |
| 1234 | + | |
1232 | 1235 | | |
1233 | 1236 | | |
1234 | 1237 | | |
| |||
0 commit comments