Go
pytorch
fe1232a4
- Use FastAtomicAdd for index_add/_index_reduce
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
2 years ago
Use FastAtomicAdd for index_add/_index_reduce Pull Request resolved: https://github.com/pytorch/pytorch/pull/76996 Approved by: https://github.com/cpuhrsch, https://github.com/ngimel
Author
mikaylagawarecki
Committer
pytorchmergebot
Parents
906fc777
Loading