pytorch/test/export
Zhengxu Chen 3ef2dfc1ba [export] Implement cpp deserializer. (#136398)
Differential Revision: D63206258

This diff introduces a mechanism to generate a json-compatible deserializer in cpp using nlohmann json (already being used by AOTI).

Why we need this? Because there will be a lot of cases where people don't want to use Python to load the graph (e.g. cpp runtime), and instead they can use this header to deserialize the JSON graph.

Every time we call update_schema.py to update the schema, the header will be auto generated and included into the source files.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/136398
Approved by: https://github.com/angelayi
2024-11-14 16:34:59 +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_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 [export] Dedup data-dependent errors based on stacktrace (#139540) 2024-11-05 18:16:05 +00:00
test_experimental.py Move test_experimental.py to training IR (#138140) 2024-10-17 17:30:10 +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_export.py [export] Implement cpp deserializer. (#136398) 2024-11-14 16:34:59 +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 [export] Implement cpp deserializer. (#136398) 2024-11-14 16:34:59 +00:00
test_serdes.py [export] Add serdes_non_strict to tests (#138662) 2024-10-24 21:35:32 +00:00
test_serialize.py Make test_torchbind.py training IR compatible (#138658) 2024-11-04 17:43:11 +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] Remove custom forward func in swap (#139126) 2024-10-30 16:50:57 +00:00
test_tools.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
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 [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_training_ir.py fix unflatten training ir test suffix (#138840) 2024-10-24 23:42:54 +00:00
test_unflatten.py [export] Add ability to run eagerly on UnflattenedModule (#133996) 2024-09-05 20:28:42 +00:00
test_verifier.py Move test_verifier to training IR (#138249) 2024-10-18 07:36:29 +00:00
testing.py [export] Implement cpp deserializer. (#136398) 2024-11-14 16:34:59 +00:00