pytorch
1e725c97 - Avoid device casting for all singleton tensors in optimizer states (#91454)

Commit
2 years ago
Avoid device casting for all singleton tensors in optimizer states (#91454) Fixes #75224 Pull Request resolved: https://github.com/pytorch/pytorch/pull/91454 Approved by: https://github.com/janeyx99
Author
Committer
Parents
Loading