Searched refs:nccl_factory (Results 1 – 2 of 2) sorted by relevance
30 std::shared_ptr<NcclUniqueIdFactory>* nccl_factory; variable171 if (nccl_factory == nullptr) { in SetNcclUniqueIdFactory()172 nccl_factory = new std::shared_ptr<NcclUniqueIdFactory>(); in SetNcclUniqueIdFactory()174 *nccl_factory = std::move(factory); in SetNcclUniqueIdFactory()179 return nccl_factory != nullptr ? *nccl_factory : nullptr; in GetNcclUniqueIdFactory()
334 std::shared_ptr<NcclUniqueIdFactory> nccl_factory = GetNcclUniqueIdFactory(); in RunExecutable() local335 if (nccl_factory != nullptr) { in RunExecutable()342 return nccl_factory->GetUniqueId(replicas); in RunExecutable()