Searched refs:NcclIdStore (Results 1 – 1 of 1) sorted by relevance
176 class NcclIdStore { class178 NcclIdStore(int node_id, std::shared_ptr<DistributedRuntimeClient> client, in NcclIdStore() function in xla::__anon0c1acb180111::NcclIdStore196 StatusOr<std::string> NcclIdStore::GetNcclUniqueId( in GetNcclUniqueId()296 auto nccl_id_store = std::make_shared<NcclIdStore>( in BuildDistributedDevices()