pytorch
3ba9e4cd - Add alltoall_ to CommTensor (#90512)

Comment changes are shownComment changes are hidden
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
Committer
Parents
  • test
    • distributed
      • File
        test_c10d_common.py
      • File
        test_c10d_nccl.py
    • forward_backward_compatibility
      • File
        check_forward_backward_compatibility.py
  • torch
    • csrc/distributed/c10d
      • File
        Ops.cpp
      • File
        Ops.hpp
      • File
        init.cpp
    • distributed/_spmd
      • File
        comm_tensor.py