Home
last modified time | relevance | path

Searched refs:onCompletionHookThread_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.hpp1028 std::thread onCompletionHookThread_; member in c10d::ProcessGroupNCCL
H A DProcessGroupNCCL.cpp1048 onCompletionHookThread_ = std::thread(&ProcessGroupNCCL::runHookLoop, this); in registerOnCompletionHook()
1275 if (onCompletionHookThread_.joinable()) { in ~ProcessGroupNCCL()
1276 onCompletionHookThread_.join(); in ~ProcessGroupNCCL()