pytorch
ff96f6d0 - [core IR][reland] Add `split.Tensor` and `unbind` decompositions to core ATen decomp table (#110323)

Commit
1 year ago
[core IR][reland] Add `split.Tensor` and `unbind` decompositions to core ATen decomp table (#110323) Summary: This is a reland of [github PR #110102]( https://github.com/pytorch/pytorch/pull/110102). The original PR had to be unlanded due to internal CI failures. This diff applies some small fixes to the failing tests to adjust to the new decompositions. Note that `lift_fresh` will not be decomposed for now, since it was found that [constant propogation looks specifically for `lift_fresh`](https://github.com/pytorch/pytorch/blob/13af952f94e611f67ac306233898b753a9cf73d2/torch/fx/experimental/proxy_tensor.py#L381-L386). Therefore decomposing `lift_fresh` will interfere with constant propogation during export. Test Plan: Github CI and internal CI Differential Revision: D49761321 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110323 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading