Searched refs:ProcessGroupMPI (Results 1 – 9 of 9) sorted by relevance
101 std::vector<at::Tensor> ProcessGroupMPI::WorkMPI::result() { in result()105 c10::intrusive_ptr<c10::ivalue::Future> ProcessGroupMPI::WorkMPI::getFuture() { in getFuture()109 void ProcessGroupMPI::WorkMPI::finishWorkMPIError( in finishWorkMPIError()115 void ProcessGroupMPI::WorkMPI::finishWorkMPI() { in finishWorkMPI()120 ProcessGroupMPI::AsyncWork::AsyncWork( in AsyncWork()131 ProcessGroupMPI::AsyncWork::~AsyncWork() { in ~AsyncWork()140 bool ProcessGroupMPI::AsyncWork::isCompleted() { in isCompleted()161 bool ProcessGroupMPI::AsyncWork::isSuccess() const { in isSuccess()171 int ProcessGroupMPI::AsyncWork::sourceRank() const { in sourceRank()175 bool ProcessGroupMPI::AsyncWork::wait(std::chrono::milliseconds /* unused */) { in wait()[all …]
82 class TORCH_API ProcessGroupMPI : public Backend { class101 friend class ProcessGroupMPI;144 explicit ProcessGroupMPI(int rank, int size, MPI_Comm pgComm);146 ~ProcessGroupMPI() override;232 static c10::intrusive_ptr<ProcessGroupMPI> createProcessGroupMPI(
2843 intrusive_ptr_no_gil_destructor_class_<::c10d::ProcessGroupMPI>( in c10d_init()2852 return ::c10d::ProcessGroupMPI::createProcessGroupMPI(std::move(ranks)); in c10d_init()
17 c10::intrusive_ptr<::c10d::ProcessGroupMPI> pg, in waitWork()34 c10::intrusive_ptr<::c10d::ProcessGroupMPI> pg, in waitFuture()58 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testAllreduce()89 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testBroadcast()121 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testReduce()152 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testAllgather()191 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testGather()241 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testScatter()283 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testSendRecv()342 auto pg = c10d::ProcessGroupMPI::createProcessGroupMPI(); in testBackendName()
603 class ProcessGroupMPI(Backend):611 def create(ranks: list[int]) -> ProcessGroupMPI: ...
165 from torch._C._distributed_c10d import ProcessGroupMPI167 ProcessGroupMPI.__module__ = "torch.distributed.distributed_c10d"1728 backend_class = ProcessGroupMPI.create(global_ranks_in_group)
163 `ProcessGroupGloo`, `ProcessGroupNCCL`, and `ProcessGroupMPI`.
1369 "${TORCH_SRC_DIR}/csrc/distributed/c10d/ProcessGroupMPI.cpp"
504 "torch/csrc/distributed/c10d/ProcessGroupMPI.cpp",