Home
last modified time | relevance | path

Searched refs:CollectiveInitializer (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/
Dcollective_init.cc23 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()
Dcollective_init.h35 class CollectiveInitializer {
37 CollectiveInitializer(CollectiveInitializer const &) = delete;
38 CollectiveInitializer &operator=(const CollectiveInitializer &) = delete;
39 static CollectiveInitializer &instance();
46 CollectiveInitializer() : collective_inited_(false) {} in CollectiveInitializer() function
47 ~CollectiveInitializer() = default;
/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc31 using CollectiveInitializer = mindspore::device::gpu::CollectiveInitializer; typedef
134 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()
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc66 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()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_send_gpu_kernel.h77 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
Dnccl_recv_gpu_kernel.h81 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
Dsync_batch_norm_grad_gpu_kernel.h124 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
Dnccl_collective_gpu_kernel.h133 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
Dsync_batch_norm_gpu_kernel.h140 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc97 using CollectiveInitializer = device::gpu::CollectiveInitializer; typedef
102 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in Init()
103 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in Init()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc123 collective_handle_ = device::gpu::CollectiveInitializer::instance().collective_handle(); in LaunchAllReduce()
Dgpu_kernel_runtime.cc98 const void *collective_handle_ = CollectiveInitializer::instance().collective_handle(); in Init()
99 bool collective_inited = CollectiveInitializer::instance().collective_inited(); in Init()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc333 (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()