You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Update base for Update on "[ET-VK][13/n] split_with_sizes with more test codegen"
Life is fun when the code-gen is more challenging than writing the operator itself.
1. Test codegen update to include vector of Tensor as output: `Tensor(a)[]`.
2. `aten.split_with_sizes.default`
3. `aten.split.Tensor`
4. Improve `DimUtils` for better dimension reasoning.
Differential Revision: [D56660525](https://our.internmc.facebook.com/intern/diff/D56660525/)
[ghstack-poisoned]
0 commit comments