Home
last modified time | relevance | path

Searched defs:WorkNCCL (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.hpp262 class WorkNCCL : public Work, public std::enable_shared_from_this<WorkNCCL> { class in c10d::ProcessGroupNCCL
H A DProcessGroupNCCL.cpp449 ProcessGroupNCCL::WorkNCCL::WorkNCCL( in WorkNCCL() function in c10d::ProcessGroupNCCL::WorkNCCL
488 ProcessGroupNCCL::WorkNCCL::WorkNCCL(const WorkNCCL& w) in WorkNCCL() function in c10d::ProcessGroupNCCL::WorkNCCL