Searched refs:NcclUniqueIdFactory (Results 1 – 3 of 3) sorted by relevance
42 struct NcclUniqueIdFactory { struct43 virtual ~NcclUniqueIdFactory() {} in ~NcclUniqueIdFactory() argument49 void SetNcclUniqueIdFactory(std::shared_ptr<NcclUniqueIdFactory> factory);51 std::shared_ptr<NcclUniqueIdFactory> GetNcclUniqueIdFactory();
30 std::shared_ptr<NcclUniqueIdFactory>* nccl_factory;169 void SetNcclUniqueIdFactory(std::shared_ptr<NcclUniqueIdFactory> factory) { in SetNcclUniqueIdFactory()172 nccl_factory = new std::shared_ptr<NcclUniqueIdFactory>(); in SetNcclUniqueIdFactory()177 std::shared_ptr<NcclUniqueIdFactory> GetNcclUniqueIdFactory() { in GetNcclUniqueIdFactory()
334 std::shared_ptr<NcclUniqueIdFactory> nccl_factory = GetNcclUniqueIdFactory(); in RunExecutable()