Home
last modified time | relevance | path

Searched refs:enable_mpi (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/parallel/mpi/
D_mpi_config.py51 def enable_mpi(self): member in _MpiConfig
54 @enable_mpi.setter
55 def enable_mpi(self, enable_mpi): member in _MpiConfig
56 self._mpiconfig_handle.set_enable_mpi(enable_mpi)
74 @args_type_check(enable_mpi=bool)
/third_party/mindspore/mindspore/ccsrc/utils/mpi/
Dmpi_config.h32 bool enable_mpi() const { return enable_mpi_; } in enable_mpi() function
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc132 .def("get_enable_mpi", &mindspore::MpiConfig::enable_mpi, "Get whether enable mpi.") in PYBIND11_MODULE()