pytorch/torch/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
..
experimental PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
passes PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
__init__.py Use typing.IO[bytes] instead of io.BytesIO in annotations (#144994) 2025-01-27 18:08:07 +00:00
_draft_export.py [export][dynamic shapes] log provenance for locals & symbols for non-strict (#143378) 2025-02-07 05:46:05 +00:00
_remove_auto_functionalized_pass.py remove allow-untyped-defs from torch/export/_remove_auto_functionalized_pass.py (#144230) 2025-01-06 22:23:19 +00:00
_remove_effect_tokens_pass.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_safeguard.py
_swap.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_trace.py move and fix logic to update unbacked bindings (#146115) 2025-02-07 22:41:19 +00:00
_tree_utils.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_unlift.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
custom_obj.py
decomp_utils.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
dynamic_shapes.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
exported_program.py Fix constants with non-functional operators (#145593) 2025-02-05 17:44:19 +00:00
graph_signature.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
unflatten.py [export] Include metadata in FlatArgsAdapter (#146107) 2025-02-05 00:29:58 +00:00