Files
pytorch/test/export
Tugsbayasgalan Manlaibaatar bb1631973d Support closure variables in nested functiuon defs (#170705)
Changelogs:
- Need to update some vmap tests because as we rebuild nested function, we need to realize the values from closure which affects graph input ordering
- Need to add some state to nested function to break cycles
- When closure refers to nested function itself, break cycle gracefully
- Subclassed NotImplementedError to catch exception in export better.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/170705
Approved by: https://github.com/williamwen42
2025-12-22 17:04:45 +00:00
..