Avik Chaudhuri 711c8c821e shape guards (#161178) ...

Summary: This PR introduces shape guards to export. Previously only value ranges,  equalities, and specializations would be tracked for symbolic expressions, and we had a forward hook to check them. Instead now we create a function to check shape guards and call it in the exported program.

Test Plan:
updated several tests

Rollback Plan:

Differential Revision: D80713603

Pull Request resolved: https://github.com/pytorch/pytorch/pull/161178
Approved by: https://github.com/tugsbayasgalan

2025-09-08 22:44:09 +00:00

..

__init__.py

…

after_aot.py

Optimize dynamo typing (#147499)

2025-08-25 13:20:45 +00:00

after_dynamo.py

typing registry.py (#160367)

2025-08-15 02:09:31 +00:00

aoti.py

shape guards (#161178)

2025-09-08 22:44:09 +00:00