pytorch/test/export
Stephen Jia ff96f6d04f [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`](13af952f94/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
2023-10-03 14:35:04 +00:00
..
test_db.py deprecate constraints in favor of dynamic_shapes (#110143) 2023-09-28 10:26:21 +00:00
test_export.py [export] Add run_decomposition() function to ExportedProgram (#110236) 2023-10-01 18:18:27 +00:00
test_funtionalized_assertions.py
test_pass_infra.py [export] Change _generate_new_graph_signature (#108571) 2023-09-06 21:39:26 +00:00
test_passes.py deprecate constraints in favor of dynamic_shapes (#110143) 2023-09-28 10:26:21 +00:00
test_serialize.py [core IR][reland] Add split.Tensor and unbind decompositions to core ATen decomp table (#110323) 2023-10-03 14:35:04 +00:00
test_upgrade.py Hide transform method by renaming it (#107940) 2023-08-25 16:31:44 +00:00
test_verifier.py [export] Verifier for exported program (#109519) 2023-09-26 18:47:43 +00:00