Go
pytorch
1e725c97
- Avoid device casting for all singleton tensors in optimizer states (#91454)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year 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
jbschlosser
Committer
pytorchmergebot
Parents
97925506
Loading