Searched refs:NcclClique (Results 1 – 4 of 4) sorted by relevance
90 class NcclClique {92 explicit NcclClique(114 LockedNcclClique(NcclClique& clique, std::unique_ptr<absl::MutexLock> lock,119 NcclClique& clique;131 StatusOr<NcclClique*> GetOrTryCreateIfAbsent(133 const std::function<StatusOr<std::unique_ptr<NcclClique>>(138 const std::function<void(const NcclCliqueKey&, const NcclClique&)>& fn)143 absl::flat_hash_map<NcclCliqueKey, std::unique_ptr<NcclClique>> map_
100 NcclClique::NcclClique( in NcclClique() function in xla::gpu::NcclClique104 ncclComm_t NcclClique::GetCommForDeviceOrdinal(int device_ordinal) const { in GetCommForDeviceOrdinal()150 StatusOr<std::unique_ptr<NcclClique>> CreateNcclClique( in CreateNcclClique()207 return std::make_unique<NcclClique>(std::move(comms_by_device_ordinal)); in CreateNcclClique()238 TF_ASSIGN_OR_RETURN(NcclClique * clique, maybe_clique_); in RunCollectiveOp()252 StatusOr<NcclClique*> maybe_clique_;288 LockedNcclClique::LockedNcclClique(NcclClique& clique, in LockedNcclClique()308 StatusOr<NcclClique*> NcclCliqueMap::GetOrTryCreateIfAbsent( in GetOrTryCreateIfAbsent()310 const std::function<StatusOr<std::unique_ptr<NcclClique>>( in GetOrTryCreateIfAbsent()315 TF_ASSIGN_OR_RETURN(std::unique_ptr<NcclClique> value, value_factory(key)); in GetOrTryCreateIfAbsent()[all …]
27 NcclCliqueCache().ForEach([&](const NcclCliqueKey& k, const NcclClique&) { in DevicesWithOpenNcclChannels() argument
59 struct NcclClique;