Searched defs:ncclStream (Results 1 – 1 of 1) sorted by relevance
246 at::cuda::CUDAStream& ncclStream) { in syncStream()2608 auto ncclStream = ncclStreams_.at(key); in checkForNCCLErrorsInternal() local2706 auto ncclStream = ncclStreams_.at(key); in checkForNCCLErrorsInternal() local2891 auto ncclStream = ncclStreams_.at(key); in checkForNCCLErrorsInternal() local3111 auto ncclStream = ncclStreams_.at(key); in checkForNCCLErrorsInternal() local3388 [&](at::cuda::CUDAStream& ncclStream, in checkForNCCLErrorsInternal()3389 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) { in checkForNCCLErrorsInternal()3795 [&](at::cuda::CUDAStream& ncclStream, in checkForNCCLErrorsInternal()3796 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) { in checkForNCCLErrorsInternal()3917 c10::intrusive_ptr<ProcessGroupNCCL::WorkNCCL>& work) { in checkForNCCLErrorsInternal()