pytorch/torch/_export/serde
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
..
gen-cpp2 [export][ez] Fix forward D67044185 (#143193) 2024-12-13 16:06:42 +00:00
__init__.py
aoti_schema.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
dynamic_shapes.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
export_schema.thrift [export] Additionally save pytree namedtuple field names (#145956) 2025-02-04 04:42:30 +00:00
schema.py [export] Additionally save pytree namedtuple field names (#145956) 2025-02-04 04:42:30 +00:00
schema.yaml [export] Additionally save pytree namedtuple field names (#145956) 2025-02-04 04:42:30 +00:00
schema_check.py [export][ez] Fix generated header file. (#146208) 2025-02-03 06:01:05 +00:00
serialize.py move and fix logic to update unbacked bindings (#146115) 2025-02-07 22:41:19 +00:00
union.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00