Go
pytorch
855f611b
- [spmd] skip gradient copying for fused adam (#99489)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
[spmd] skip gradient copying for fused adam (#99489) gradients does not need to be copy back as it's not useful Pull Request resolved: https://github.com/pytorch/pytorch/pull/99489 Approved by: https://github.com/mrshenli
Author
wanchaol
Committer
pytorchmergebot
Parents
ca24a962
Loading