Commit 9343d28
ssjia
[ET-VK][ez] Fix IndexError in Vulkan partitioner DtypeSetList/TensorRepSetList
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.
Differential Revision: [D95970163](https://our.internmc.facebook.com/intern/diff/D95970163/)
[ghstack-poisoned]1 parent f09bd55 commit 9343d28
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