pytorch
bd1ffc65 - [Dynamo] Fix bug: GradMode doesn't carry grad state correctly after graph break (#88537)

Commit
3 years ago
[Dynamo] Fix bug: GradMode doesn't carry grad state correctly after graph break (#88537) Fixes https://github.com/pytorch/torchdynamo/issues/1446 Pull Request resolved: https://github.com/pytorch/pytorch/pull/88537 Approved by: https://github.com/jansel
Author
Committer
Parents
Loading