Home
last modified time | relevance | path

Searched refs:ProcessGroupWrapper (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupWrapper.cpp385 ProcessGroupWrapper::ProcessGroupWrapper( in ProcessGroupWrapper() function in c10d::ProcessGroupWrapper
395 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 …]
H A DProcessGroupWrapper.hpp11 class TORCH_API ProcessGroupWrapper : public Backend { class
13 explicit ProcessGroupWrapper(
H A Dinit.cpp2648 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()
/aosp_15_r20/external/pytorch/
H A Dbuild_variables.bzl505 "torch/csrc/distributed/c10d/ProcessGroupWrapper.cpp",