pytorch
c2e7a0d6 - [core IR] Add decomps for `aten.sum` and `aten.squeeze` variants (#110645)

Commit
1 year ago
[core IR] Add decomps for `aten.sum` and `aten.squeeze` variants (#110645) Summary: ## Context Both `aten.sum` and `aten.squeeze` have a "most generic" variant in the form of `aten.sum.dim_IntList` and `aten.squeeze.dims` respectively. Add decompositions for other non generic variants of these operators to express them using the most generic variant. Note that to register these decomps, the reference implementation under `_refs` had to be removed as registered decompositions. cc: @lezcano @peterbell10 Test Plan: Github CI + Meta Internal CI Differential Revision: D49965952 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110645 Approved by: https://github.com/peterbell10, https://github.com/digantdesai, https://github.com/manuelcandales
Author
Committer
Parents
Loading