pytorch
3ba9e4cd
- Add alltoall_ to CommTensor (#90512)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Comment Changes
Previous Change (CTRL+↑)
Next Change (CTRL+↓)
Expand Context Lines
Collapse Context Lines
Hide Minimap (CTRL+M)
Commit
2 years ago
Add alltoall_ to CommTensor (#90512) This PR adds alltoall_ to the CommTensor Pull Request resolved: https://github.com/pytorch/pytorch/pull/90512 Approved by: https://github.com/mrshenli
Author
wanchaol
Committer
pytorchmergebot
Parents
6165a180
Files
7
test
distributed
test_c10d_common.py
test_c10d_nccl.py
forward_backward_compatibility
check_forward_backward_compatibility.py
torch
csrc/distributed/c10d
Ops.cpp
Ops.hpp
init.cpp
distributed/_spmd
comm_tensor.py
Loading