2 Commits

Author	SHA1	Message	Date
Tugsbayasgalan Manlaibaatar	825fe15024	EZ fix to make sure local pytest run succeeds in export (#144764) ... Previously run_tests() was protected under IS_FBCODE flag so that following works: ``` python test/export/test_export_legacy.py ``` But it fails on: ``` pytest test/export/test_export_legacy.py ``` This is because pytest doesn't seem to get triggered through run_tests(). Differential Revision: [D68152737](https://our.internmc.facebook.com/intern/diff/D68152737) Pull Request resolved: https://github.com/pytorch/pytorch/pull/144764 Approved by: https://github.com/avikchaudhuri	2025-01-15 00:43:40 +00:00
Tugsbayasgalan Manlaibaatar	87f9c1abe5	Change export IR to non-functional pre-dispatch IR (#139511) ... Differential Revision: [D65362160](https://our.internmc.facebook.com/intern/diff/D65362160) State after this IR: 1. For the tests that require inference IR, they are replaced with ep.run_decomp({}) so export_for_training_run_decomp is sort of redundant but i guess it is still nice that multiple round of retracing still working. In general, we need some auditing to reduce our redundant testing coverages. 2. After this PR landed and not get reverted for a week or so, i will replace the export_for_training calls with export as they are the same thing now. 3. Added more tests to also cover now "deprecated" old IR by patching export to use old export. For reviewers, please look at the internal version. Pull Request resolved: https://github.com/pytorch/pytorch/pull/139511 Approved by: https://github.com/ydwu4, https://github.com/angelayi, https://github.com/avikchaudhuri	2024-11-20 21:47:55 +00:00