Home
last modified time | relevance | path

Searched refs:GetRankID (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dcomm_manager.cc88 bool CommManager::GetRankID(const string &group, unsigned int *rank_id) const { in GetRankID() function in mindspore::CommManager
150 bool CommManager::GetRankID(const string &group, unsigned int *rank_id) const {
202 bool CommManager::GetRankID(const string &group, unsigned int *rank_id) const { return true; }
242 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRank()
Dcomm_manager.h36 bool GetRankID(const string &group, unsigned int *rank_id) const;
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/
Ddevice_context.h114 virtual uint32_t GetRankID() const { return 0; } in GetRankID() function
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.h68 uint32_t GetRankID() const override;
Dgpu_device_context.cc102 auto rank_id = GetRankID(); in Initialize()
497 uint32_t GPUDeviceContext::GetRankID() const { in GetRankID() function in mindspore::device::gpu::GPUDeviceContext
502 if (!CommManager::GetInstance().GetRankID(kNcclWorldGroup, &rank_id)) { in GetRankID()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dgroup_manager.h61 Status GetRankID(const std::string &name, uint32_t *rank_id);
Dgroup_manager.cc249 Status GroupManager::GetRankID(const std::string &name, uint32_t *const rank_id) { in GetRankID() function in mindspore::parallel::GroupManager
255 bool ret = CommManager::GetInstance().GetRankID(name, rank_id); in GetRankID()
Dstep_parallel.cc2915 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetCommInfo()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_split.cc50 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRank()
Dpipeline.cc1334 bool ret = CommManager::GetInstance().GetRankID("", &rank_id); in GetHcclRankId()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger.cc380 uint32_t Debugger::GetRankID() { in GetRankID() function in mindspore::Debugger
387 uint32_t rank_id = device_context->GetRankID(); in GetRankID()
392 uint32_t rank_id = GetRankID(); in Dump()
403 uint32_t rank_id = GetRankID(); in DumpSingleNode()
419 uint32_t rank_id = GetRankID(); in DumpInGraphCompiler()
Ddebugger.h89 static uint32_t GetRankID();
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc60 auto rank_id = GetRankID(); in Initialize()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/
Ddata_dumper.cc156 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in SetOpMappingInfo()
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Ddump_json_parser.cc560 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in GetOpOverflowBinPath()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc658 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in GetDumpPath()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc2744 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRankId()