pytorch
f30694c7
- Add allgather_into_tensor to CommTensor (#90565)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Add allgather_into_tensor to CommTensor (#90565) This PR adds _all_gather_base_ to CommTensor to support allgather_base Pull Request resolved: https://github.com/pytorch/pytorch/pull/90565 Approved by: https://github.com/mrshenli
Author
wanchaol
Committer
pytorchmergebot
Parents
b782927e
Loading