pytorch/test/export
Pian Pawakapan b897ab0540 [export] ignore mark_dynamic() in export (#135536)
Previously we were accomodating `torch._dynamo.mark_dynamic()` for export's dynamic shapes. Here we clean things up and ignore it, requiring users to specify an export input for `dynamic_shapes`.

Note: there's 4 decorators relevant to export, `mark_dynamic, maybe_mark_dynamic, mark_static, mark_unbacked`. User calls that involve export have only been `mark_dynamic()`, and we use `maybe_mark_dynamic` under the hood for `Dim.AUTO`, but we could start using others. One reason I decided to not warn and just silently ignore is these decorators cause the tensors to carry dynamic info, and it'll be hard to tell whether the markers are from export or user calls when re-exporting with the same inputs.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/135536
Approved by: https://github.com/avikchaudhuri
2024-09-12 21:22:19 +00:00
..
__init__.py [export] Fix test to run internally (#116118) 2023-12-20 01:02:16 +00:00
opinfo_schema.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_converter.py [ts_converter][reland] Add support for LinearOpContext and Conv2dOpContext in quantization pass (#133622) 2024-08-16 01:55:45 +00:00
test_db.py [export] Update example inputs format for DB. (#129982) 2024-07-03 17:53:15 +00:00
test_experimental.py Fix decomp behaviour in export training IR (#134801) 2024-09-05 06:37:44 +00:00
test_export.py [export] ignore mark_dynamic() in export (#135536) 2024-09-12 21:22:19 +00:00
test_export_nonstrict.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_export_training_ir_to_run_decomp.py Add export_for_training as public API (#134677) 2024-08-28 22:32:10 +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 [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_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 [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_serialize.py Back out "[pytorch][PR] [export] Schematize nn_module_stack serialization" (#134628) 2024-08-28 03:45:46 +00:00
test_sparse.py [traced-graph][sparse] enable to_dense() for compressed (#133371) 2024-08-24 20:33:23 +00:00
test_tools.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_torchbind.py Make the __module__ name of HOO to be always "torch.ops.higher_order" (#132775) 2024-08-08 16:55:09 +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_verifier.py [export] Publicize validate function (#132777) 2024-08-07 23:10:05 +00:00
testing.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