Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/utils/mpi/
Dmpi_config.h31 void set_enable_mpi(bool flag) { enable_mpi_ = flag; } in set_enable_mpi() function
/third_party/mindspore/mindspore/parallel/mpi/
D_mpi_config.py56 self._mpiconfig_handle.set_enable_mpi(enable_mpi)
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc133 .def("set_enable_mpi", &mindspore::MpiConfig::set_enable_mpi, "Set whether to enable mpi."); in PYBIND11_MODULE()