pytorch/test/export
Avik Chaudhuri 103c8b44bc move and fix logic to update unbacked bindings (#146115)
Summary:
Previously we were touching up unbacked bindings between Dynamo and AOTAutograd in strict export, but the logic had a bug: if an unbacked symint gets substituted by a backed symint, we would put the backed symint in the unbacked bindings (the check `is_symbol` was not enough here).

This PR fixes this logic, and moreover, moves it into the serializer instead, because we don't need this adjustment outside serde.

Test Plan: added test

 D68880766

Pull Request resolved: https://github.com/pytorch/pytorch/pull/146115
Approved by: https://github.com/pianpwk
2025-02-07 22:41:19 +00:00
..
__init__.py
opinfo_schema.py
random_dag.py
test_converter.py
test_cpp_serdes.py
test_db.py
test_draft_export.py [dynamic shapes][real tensor tracing] propagate unbacked hint when creating mod replacement (#146381) 2025-02-06 01:48:40 +00:00
test_experimental.py
test_export.py move and fix logic to update unbacked bindings (#146115) 2025-02-07 22:41:19 +00:00
test_export_legacy.py
test_export_nonstrict.py
test_export_training_ir_to_run_decomp.py
test_functionalized_assertions.py fix incorrect literal strings / accidental tuples (#146037) 2025-02-03 15:08:11 +00:00
test_hop.py Remove unused import in tests (#146266) 2025-02-03 03:40:18 +00:00
test_lift_unlift.py
test_pass_infra.py
test_passes.py
test_retraceability.py
test_schema.py
test_serdes.py
test_serialize.py [export] Additionally save pytree namedtuple field names (#145956) 2025-02-04 04:42:30 +00:00
test_sparse.py
test_swap.py Remove unused import in tests (#146266) 2025-02-03 03:40:18 +00:00
test_tools.py
test_torchbind.py
test_tree_utils.py fix incorrect literal strings / accidental tuples (#146037) 2025-02-03 15:08:11 +00:00
test_unflatten.py [export] Include metadata in FlatArgsAdapter (#146107) 2025-02-05 00:29:58 +00:00
test_unflatten_training_ir.py
test_verifier.py
testing.py