Searched defs:nccl_stream (Results 1 – 2 of 2) sorted by relevance
34 se::Stream* nccl_stream, in GPUDeviceContext()53 se::Stream* nccl_stream() const { return nccl_stream_; } in nccl_stream() function
108 NcclStream* nccl_stream = nullptr; member359 NcclStream* nccl_stream = nullptr; in GetCommunicator() local627 NcclStream* nccl_stream = collective->communicator->members[i].nccl_stream; in RunCollective() local671 NcclStream* nccl_stream = in RunCollective() local698 void NcclManager::LoopKernelLaunches(NcclStream* nccl_stream) { in LoopKernelLaunches()