Searched refs:NcclStream (Results 1 – 2 of 2) sorted by relevance
72 struct NcclManager::NcclStream : public core::RefCounted { struct in tensorflow::NcclManager74 NcclStream() = default;75 ~NcclStream() = default;108 NcclStream* nccl_stream = nullptr;240 for (NcclStream* nccl_stream : it.second) { in ~NcclManager()347 std::set<NcclStream*> used_streams; in GetCommunicator()359 NcclStream* nccl_stream = nullptr; in GetCommunicator()367 nccl_stream = new NcclStream(); in GetCommunicator()627 NcclStream* nccl_stream = collective->communicator->members[i].nccl_stream; in RunCollective()671 NcclStream* nccl_stream = in RunCollective()[all …]
212 struct NcclStream;242 void LoopKernelLaunches(NcclStream* stream);252 absl::flat_hash_map<se::StreamExecutor*, std::vector<NcclStream*>>