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 Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
random_dag.py [BE] fix ruff rule E226: add missing whitespace around operator in f-strings (#144415) 2025-01-08 21:55:00 +00:00
test_converter.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
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 [export] handle buffer/input mutations for joint-graph (#144806) 2025-01-16 00:22:16 +00:00
test_export.py move and fix logic to update unbacked bindings (#146115) 2025-02-07 22:41:19 +00:00
test_export_legacy.py EZ fix to make sure local pytest run succeeds in export (#144764) 2025-01-15 00:43:40 +00:00
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 PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_pass_infra.py [Codemod][AddExplicitStrictExportArg] caffe2/test (#143688) 2024-12-27 07:58:44 +00:00
test_passes.py PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_retraceability.py
test_schema.py Tweak schema_check to handle annotated builtin types (#145154) 2025-01-19 18:48:35 +00:00
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 [Codemod][AddExplicitStrictExportArg] caffe2/test (#143688) 2024-12-27 07:58:44 +00:00
test_swap.py Remove unused import in tests (#146266) 2025-02-03 03:40:18 +00:00
test_tools.py
test_torchbind.py Fix unused Python variables in test/[e-z]* (#136964) 2024-12-18 23:02:30 +00:00
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 Change export IR to non-functional pre-dispatch IR (#139511) 2024-11-20 21:47:55 +00:00