Searched refs:nccl_stream (Results 1 – 1 of 1) sorted by relevance
80 NcclStream* nccl_stream = nullptr; member231 if (comm->members[i].nccl_stream->executor != in GetCommunicator()276 NcclStream* nccl_stream = nullptr; in GetCommunicator() local279 nccl_stream = s.get(); in GetCommunicator()283 if (nccl_stream == nullptr) { in GetCommunicator()284 nccl_stream = new NcclStream(); in GetCommunicator()285 nccl_stream->executor = executor; in GetCommunicator()286 nccl_stream->stream.reset(new se::Stream(executor)); in GetCommunicator()287 nccl_stream->stream->Init(); in GetCommunicator()289 streams.emplace_back(nccl_stream); in GetCommunicator()[all …]