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
..
db PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
pass_infra PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
passes [export] make stack_trace optional in insert_custom_op_guards (#146438) 2025-02-06 01:48:26 +00:00
serde move and fix logic to update unbacked bindings (#146115) 2025-02-07 22:41:19 +00:00
__init__.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
converter.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
error.py
non_strict_utils.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
pass_base.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
tools.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
utils.py Fix constants with non-functional operators (#145593) 2025-02-05 17:44:19 +00:00
verifier.py Introduce aoti_call_delegate HOP (#145630) 2025-01-31 04:57:36 +00:00
wrappers.py