Searched refs:ProcessGroupWrapper (Results 1 – 4 of 4) sorted by relevance
385 ProcessGroupWrapper::ProcessGroupWrapper( in ProcessGroupWrapper() function in c10d::ProcessGroupWrapper395 const std::string ProcessGroupWrapper::getBackendName() const { in getBackendName()399 c10::intrusive_ptr<Work> ProcessGroupWrapper::broadcast( in broadcast()406 c10::intrusive_ptr<Work> ProcessGroupWrapper::allreduce( in allreduce()413 c10::intrusive_ptr<Work> ProcessGroupWrapper::allreduce_coalesced( in allreduce_coalesced()424 c10::intrusive_ptr<Work> ProcessGroupWrapper::reduce( in reduce()431 c10::intrusive_ptr<Work> ProcessGroupWrapper::allgather( in allgather()443 c10::intrusive_ptr<Work> ProcessGroupWrapper::_allgather_base( in _allgather_base()452 c10::intrusive_ptr<Work> ProcessGroupWrapper::allgather_coalesced( in allgather_coalesced()464 c10::intrusive_ptr<Work> ProcessGroupWrapper::gather( in gather()[all …]
11 class TORCH_API ProcessGroupWrapper : public Backend { class13 explicit ProcessGroupWrapper(
2648 intrusive_ptr_no_gil_destructor_class_<::c10d::ProcessGroupWrapper>( in c10d_init()2654 return c10::make_intrusive<::c10d::ProcessGroupWrapper>( in c10d_init()2661 "wrapped_pg", &::c10d::ProcessGroupWrapper::getWrappedPg); in c10d_init()
505 "torch/csrc/distributed/c10d/ProcessGroupWrapper.cpp",