pytorch
c9ce0e63 - [Dynamo] Support context wrapping(e.g, torch.no_grad) on nested functions w/o closure (#92922)

Commit
1 year ago
[Dynamo] Support context wrapping(e.g, torch.no_grad) on nested functions w/o closure (#92922) Fixes 14k github models: https://github.com/jansel/pytorch-jit-paritybench/blob/master/generated/test_ELEKTRONN_elektronn3.py Pull Request resolved: https://github.com/pytorch/pytorch/pull/92922 Approved by: https://github.com/jansel, https://github.com/mlazos
Author
Committer
Parents
Loading