Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/distribute/
Dmpi_pycc.cc32 int MpiPycc::GetRankId(const std::string &group) { return GetRankIdByGroup(group); } in GetRankId() function in mindspore::device::ascend::collective::MpiPycc
41 mpi_initializer.def("get_rank_id", &MpiPycc::GetRankId, "get rank id"); in PYBIND11_MODULE()
Dmpi_pycc.h34 static int GetRankId(const std::string &group);
Dascend_collective.h50 int GetRankId(const std::string &name = kHcclWorldGroup) const;
Dascend_collective.cc76 int HcclCollectiveGroup::GetRankId(const std::string &name) const { in GetRankId() function in mindspore::device::ascend::collective::HcclCollectiveGroup
/third_party/mindspore/tests/ut/cpp/stub/hccl/
Dcollective_stub.cc28 int HcclCollectiveGroup::GetRankId(const std::string &) const { return 0; } in GetRankId() function in mindspore::device::ascend::collective::HcclCollectiveGroup
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/mpi/
Dmpi_adapter.h35 FUNC_EXPORT int GetRankId() const { return rank_id_; } in GetRankId() function
Dmpi_export.cc26 return inst->GetRankId(); in GetMPIRankId()
/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc94 *rank_id = static_cast<unsigned int>(HcclCollectiveGroup::instance().GetRankId(group)); in GetRankID()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.h340 uint32_t GetRankId();
Dgpu_session.cc123 rank_id_ = GetRankId(); in Init()
Dascend_session.cc1239 rank_id_ = GetRankId(); in InitRuntimeResource()
Dsession_basic.cc2729 uint32_t GetRankId() { in GetRankId() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc98 auto rank = HcclCollectiveGroup::instance().GetRankId(); in GetRankIdStr()