Home
last modified time | relevance | path

Searched refs:get_rank_id (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/model_utils/
Ddevice_adapter.py21 from .moxing_adapter import get_device_id, get_device_num, get_rank_id, get_job_id
23 from .local_adapter import get_device_id, get_device_num, get_rank_id, get_job_id
Dlocal_adapter.py30 def get_rank_id(): function
Dmoxing_adapter.py35 def get_rank_id(): function
/third_party/mindspore/mindspore/communication/
D_comm_helper.py202 rank_id = mpi.get_rank_id(group)
205 rank_id = hccl.get_rank_id()
207 rank_id = hccl.get_rank_id(group)
209 rank_id = mpi.get_rank_id(group)
232 rank_id = mpi.get_rank_id(group)
D_hccl_management.py168 def get_rank_id(group="hccl_world_group"): function
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/mpi/
Dmpi_initializer.cc33 int MPIInitializer::get_rank_id(const std::string &group) { return GetRankIDByGroup(group); } in get_rank_id() function in mindspore::device::gpu::MPIInitializer
39 mpi_initializer.def("get_rank_id", &MPIInitializer::get_rank_id, "get rank id"); in PYBIND11_MODULE()
Dmpi_initializer.h31 static int get_rank_id(const std::string &group);
/third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/
Dmi_pybind_register.cc50 .def("get_rank_id", &watchpoint_hit_t::get_rank_id) in PYBIND11_MODULE()
58 .def("get_rank_id", &tensor_info_t::get_rank_id) in PYBIND11_MODULE()
Ddbg_services.h68 const uint32_t get_rank_id() const { return rank_id; } in get_rank_id() function
92 const uint32_t get_rank_id() const { return rank_id; } in get_rank_id() function
/third_party/mindspore/mindspore/offline_debug/
Ddbg_services.py236 rank_id = watchpoint.get_rank_id()
476 return self.instance.get_rank_id()
1227 return self.instance.get_rank_id()
/third_party/mindspore/tests/ut/python/hccl_test/manage/
Dapi.py58 def get_rank_id(group=None): function