Searched refs:nccl_stream (Results 1 – 2 of 2) sorted by relevance
108 NcclStream* nccl_stream = nullptr; member240 for (NcclStream* nccl_stream : it.second) { in ~NcclManager()242 mutex_lock l(nccl_stream->mu); in ~NcclManager()243 nccl_stream->shutdown_requested = true; in ~NcclManager()244 nccl_stream->cv.notify_all(); in ~NcclManager()246 nccl_stream->Unref(); in ~NcclManager()314 if (comm->members[i].nccl_stream->executor != in GetCommunicator()359 NcclStream* nccl_stream = nullptr; in GetCommunicator() local362 nccl_stream = s; in GetCommunicator()366 if (nccl_stream == nullptr) { in GetCommunicator()[all …]
34 se::Stream* nccl_stream, in GPUDeviceContext() argument42 nccl_stream_(nccl_stream), in GPUDeviceContext()53 se::Stream* nccl_stream() const { return nccl_stream_; } in nccl_stream() function