Searched defs:nccl_stream (Results 1 – 1 of 1) sorted by relevance
80 NcclStream* nccl_stream = nullptr; member276 NcclStream* nccl_stream = nullptr; in GetCommunicator() local505 NcclStream* nccl_stream = collective->communicator->members[i].nccl_stream; in RunCollective() local532 NcclStream* nccl_stream = in RunCollective() local541 void NcclManager::LoopKernelLaunches(NcclStream* nccl_stream) { in LoopKernelLaunches()