pytorch/test/export
Tugsbayasgalan Manlaibaatar b7524b05d2 Make test_export training IR compatible (#138517)
In this PR, I make test_export to be compatible with training IR. The idea is that when we flip the IR to non-functional training IR, all these tests should be green. The changes involve reading through the test case, and add necessary decomposition etc to make sure the tests pass. For example, if the tests expect to see mutated buffers returned, we need to get them via running run_decomp.

Differential Revision: [D64732360](https://our.internmc.facebook.com/intern/diff/D64732360)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138517
Approved by: https://github.com/avikchaudhuri
2024-10-28 22:38:19 +00:00
..
__init__.py
opinfo_schema.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_converter.py [ts_converter] Fix prim::If buffer names (#136648) 2024-10-02 00:07:47 +00:00
test_db.py Move test_db to training IR (#138251) 2024-10-18 07:42:13 +00:00
test_experimental.py Move test_experimental.py to training IR (#138140) 2024-10-17 17:30:10 +00:00
test_export.py Make test_export training IR compatible (#138517) 2024-10-28 22:38:19 +00:00
test_export_nonstrict.py Fix lint errors in fbcode (#135614) 2024-09-13 02:04:34 +00:00
test_export_training_ir_to_run_decomp.py Remove presere ops (#138371) 2024-10-25 19:13:55 +00:00
test_functionalized_assertions.py [export] fix test ownership (#117886) 2024-01-21 01:18:16 +00:00
test_hop.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_lift_unlift.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_pass_infra.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_passes.py Rewrite unsafe_remove_auto_functionalized_pass using decompose_auto_functionalized (#134831) 2024-08-30 16:27:53 +00:00
test_retraceability.py [export] Add retraceability_non_strict to tests (#138380) 2024-10-22 21:05:51 +00:00
test_schema.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_serdes.py [export] Add serdes_non_strict to tests (#138662) 2024-10-24 21:35:32 +00:00
test_serialize.py Move test_serialize to training IR (#138261) 2024-10-23 00:32:32 +00:00
test_sparse.py [traced-graph][sparse] enable to_dense() for compressed (#133371) 2024-08-24 20:33:23 +00:00
test_swap.py [export] Move swap to a different file (#137134) 2024-10-06 04:28:18 +00:00
test_tools.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_torchbind.py [torchbind] Add generic __deepcopy__ method (#137613) 2024-10-24 22:14:55 +00:00
test_tree_utils.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00
test_unflatten.py [export] Add ability to run eagerly on UnflattenedModule (#133996) 2024-09-05 20:28:42 +00:00
test_unflatten_training_ir.py fix unflatten training ir test suffix (#138840) 2024-10-24 23:42:54 +00:00
test_verifier.py Move test_verifier to training IR (#138249) 2024-10-18 07:36:29 +00:00
testing.py [export] Add serdes_non_strict to tests (#138662) 2024-10-24 21:35:32 +00:00