Searched refs:CollectiveInitializer (Results 1 – 13 of 13) sorted by relevance
23 CollectiveInitializer &CollectiveInitializer::instance() { in instance()24 static CollectiveInitializer instance = {}; in instance()28 bool CollectiveInitializer::collective_inited() const { return collective_inited_; } in collective_inited()30 const void *CollectiveInitializer::collective_handle() const { return collective_handle_; } in collective_handle()32 void CollectiveInitializer::InitCollective() { in InitCollective()48 CollectiveInitializer::instance().collective_inited_ = true; in InitCollective()49 CollectiveInitializer::instance().collective_handle_ = handle; in InitCollective()52 void CollectiveInitializer::FinalizeCollective() { in FinalizeCollective()53 if (CollectiveInitializer::instance().collective_handle_ != nullptr) { in FinalizeCollective()54 if (dlclose(CollectiveInitializer::instance().collective_handle_) != 0) { in FinalizeCollective()
35 class CollectiveInitializer {37 CollectiveInitializer(CollectiveInitializer const &) = delete;38 CollectiveInitializer &operator=(const CollectiveInitializer &) = delete;39 static CollectiveInitializer &instance();46 CollectiveInitializer() : collective_inited_(false) {} in CollectiveInitializer() function47 ~CollectiveInitializer() = default;
31 using CollectiveInitializer = mindspore::device::gpu::CollectiveInitializer; typedef134 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle();151 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle();165 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle();179 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle();237 if (!CollectiveInitializer::instance().collective_inited()) { in GetRank()
66 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in Initialize()67 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in Initialize()498 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in GetRankID()499 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in GetRankID()
77 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
81 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
124 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
133 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
140 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
97 using CollectiveInitializer = device::gpu::CollectiveInitializer; typedef102 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in Init()103 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in Init()
123 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in LaunchAllReduce()
98 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in Init()99 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in Init()
333 (void)m.def("init_gpu_collective", &mindspore::device::gpu::CollectiveInitializer::InitCollective, in PYBIND11_MODULE()335 …(void)m.def("finalize_gpu_collective", &mindspore::device::gpu::CollectiveInitializer::FinalizeCol… in PYBIND11_MODULE()