Searched refs:MpiConfig (Results 1 – 4 of 4) sorted by relevance
23 class MpiConfig {25 ~MpiConfig() = default;26 MpiConfig(const MpiConfig &) = delete;27 MpiConfig &operator=(const MpiConfig &) = delete;29 static std::shared_ptr<MpiConfig> GetInstance();35 MpiConfig() : enable_mpi_(false) {} in MpiConfig() function37 static std::shared_ptr<MpiConfig> instance_;
20 std::shared_ptr<MpiConfig> MpiConfig::instance_ = nullptr;22 std::shared_ptr<MpiConfig> MpiConfig::GetInstance() { in GetInstance()25 instance_.reset(new (std::nothrow) MpiConfig()); in GetInstance()
19 from mindspore._c_expression import MpiConfig35 self._mpiconfig_handle = MpiConfig.get_instance()
130 (void)py::class_<mindspore::MpiConfig, std::shared_ptr<mindspore::MpiConfig>>(m, "MpiConfig") in PYBIND11_MODULE()131 .def_static("get_instance", &mindspore::MpiConfig::GetInstance, "Get mpi config instance.") in PYBIND11_MODULE()132 .def("get_enable_mpi", &mindspore::MpiConfig::enable_mpi, "Get whether enable mpi.") in PYBIND11_MODULE()133 .def("set_enable_mpi", &mindspore::MpiConfig::set_enable_mpi, "Set whether to enable mpi."); in PYBIND11_MODULE()