Zhengxu Chen a8a570512b [export] Generate compatible thrift schema out of schema.py (#141611) ... Summary: To make sure schema.py and schema.thrift are kept in sync, we use the int keys from thrift and use Python Annotated type to associate fields between thrift and schema.py. Later we will use this association to build a single source of truth between the schemas. Test Plan: CI Differential Revision: D66253157 Pull Request resolved: https://github.com/pytorch/pytorch/pull/141611 Approved by: https://github.com/yiming0416		2024-11-29 20:09:49 +00:00
..
__init__.py
opinfo_schema.py
test_converter.py
test_cpp_serdes.py	[export] Implement cpp deserializer. (#136398)	2024-11-14 16:34:59 +00:00
test_db.py	Move test_db to training IR (#138251)	2024-10-18 07:42:13 +00:00
test_draft_export.py	Fix AttributeError: 'int' object has no attribute 'node' due to constant prop (#141250)	2024-11-24 08:20:04 +00:00
test_experimental.py	[export] fix loss_output in joint graph signature (#140974)	2024-11-21 23:57:07 +00:00
test_export.py	Revert #141066 and #141494 (#141721)	2024-11-28 20:18:19 +00:00
test_export_legacy.py	Change export IR to non-functional pre-dispatch IR (#139511)	2024-11-20 21:47:55 +00:00
test_export_nonstrict.py
test_export_training_ir_to_run_decomp.py	Remove presere ops (#138371)	2024-10-25 19:13:55 +00:00
test_functionalized_assertions.py
test_hop.py
test_lift_unlift.py
test_pass_infra.py
test_passes.py	[export] avoid name collision when inlining node (#141169)	2024-11-22 01:08:22 +00:00
test_retraceability.py	[export] Add retraceability_non_strict to tests (#138380)	2024-10-22 21:05:51 +00:00
test_schema.py	[export] Generate compatible thrift schema out of schema.py (#141611)	2024-11-29 20:09:49 +00:00
test_serdes.py	[export] Add serdes_non_strict to tests (#138662)	2024-10-24 21:35:32 +00:00
test_serialize.py	Revert "Add truediv support in export serializer (#136364)"	2024-11-27 03:24:31 +00:00
test_sparse.py
test_swap.py	[export] Remove custom forward func in swap (#139126)	2024-10-30 16:50:57 +00:00
test_tools.py
test_torchbind.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-11-05 23:10:38 +00:00
test_tree_utils.py
test_unflatten.py	[export] Make unflattened module compileable (#141249)	2024-11-23 18:46:01 +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	Change export IR to non-functional pre-dispatch IR (#139511)	2024-11-20 21:47:55 +00:00