mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
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