Tarun Karuturi ff25dfca5a Save quantization_tag in export graph serialization (#127473) ... Summary: `quantization_tag` is a first class citizen metadata in quantization flows that is preserved by it. As we'll want to store the quantized exported graphs we also need to preserve this metadata as it's used in later flows. Only json supported metadata will be allowed to be serialized. Test Plan: Added test case Differential Revision: D57939282 Pull Request resolved: https://github.com/pytorch/pytorch/pull/127473 Approved by: https://github.com/angelayi		2024-07-12 05:06:40 +00:00
..
__init__.py
opinfo_schema.py	[export] add SchemaCheckMode testing for pre-dispatch export, OpInfo (#125481)	2024-05-14 21:07:21 +00:00
test_converter.py	[ts-migration] support prim::SetAttr and fix prim::GetAttr (#129440)	2024-06-29 05:08:13 +00:00
test_db.py	[export] Update example inputs format for DB. (#129982)	2024-07-03 17:53:15 +00:00
test_experimental.py	[export] match fake mode for _decompose_exported_program() (#129421)	2024-06-26 05:52:31 +00:00
test_export.py	[export] fix kwargs in run_decompositions() for training IR (#130553)	2024-07-11 22:53:18 +00:00
test_export_nonstrict.py
test_export_training_ir_to_run_decomp.py	Taskify training IR + run_decomp flow failures (#129547)	2024-06-27 20:43:22 +00:00
test_functionalized_assertions.py
test_hop.py	[while_loop] add a simiple op_info test (#123814)	2024-04-11 19:59:04 +00:00
test_lift_unlift.py	[export] handle constant aliasing for export (#125509)	2024-05-10 00:14:37 +00:00
test_pass_infra.py	[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)	2024-05-27 14:49:57 +00:00
test_passes.py	Allow kwargs in _remove_effect_tokens_pass (#130491)	2024-07-11 19:03:19 +00:00
test_retraceability.py	[Export] Add runtime assert to non-strict export (#123681)	2024-04-18 16:13:27 +00:00
test_schema.py
test_serdes.py	View specialization (#127641)	2024-06-08 05:52:52 +00:00
test_serialize.py	Save quantization_tag in export graph serialization (#127473)	2024-07-12 05:06:40 +00:00
test_sparse.py	[traced-graph][sparse] remove redundant assert in sparse prop test (#128523)	2024-06-14 17:05:17 +00:00
test_tools.py	Tool for scouting exportability in one shot (#126471)	2024-05-18 00:10:46 +00:00
test_torchbind.py	[export] make with_effect mark op has_effect to prevent them from DCEed. (#129680)	2024-07-11 22:46:21 +00:00
test_tree_utils.py
test_unflatten.py	[export] copy sym ops when respecting call module signature (#129153)	2024-06-21 01:40:22 +00:00
test_verifier.py	Revert "[cond] inlining into one of the branches when pred is a python constant (#128709)"	2024-07-10 17:47:19 +00:00
testing.py	Taskify training IR + run_decomp flow failures (#129547)	2024-06-27 20:43:22 +00:00