Searched refs:CommManager (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | comm_manager.cc | 40 CommManager &CommManager::GetInstance() noexcept { in GetInstance()
41 static CommManager instance("hccl"); in GetInstance()
70 bool CommManager::CreateGroupSync(const string &group, const vector<unsigned int> &rank_id_list) co… in CreateGroupSync()
88 bool CommManager::GetRankID(const string &group, unsigned int *rank_id) const { in GetRankID()
104 bool CommManager::GetRankSize(const string &group, unsigned int *rank_size) const { in GetRankSize()
121 bool CommManager::DestroyGroup(const string &group) const { in DestroyGroup()
128 CommManager &CommManager::GetInstance() noexcept {
129 static CommManager instance("nccl");
133 bool CommManager::CreateGroupSync(const string &group, const vector<unsigned int> &rank_id_list) co…
150 bool CommManager::GetRankID(const string &group, unsigned int *rank_id) const {
[all …]
|
| D | comm_manager.h | 31 class CommManager {
33 static CommManager &GetInstance() noexcept;
38 ~CommManager() = default;
40 CommManager(const CommManager &) = delete;
43 explicit CommManager(string backend) : backend_(std::move(backend)) {} in CommManager() function
|
| /third_party/mindspore/tests/ut/cpp/utils/ |
| D | comm_manager_test.cc | 27 ASSERT_TRUE(CommManager::GetInstance().CreateGroupSync(string("1-2-3"), devices)); in TEST_F()
28 ASSERT_TRUE(CommManager::GetInstance().CreateGroupSync(string("hccl_world_group"), devices)); in TEST_F()
33 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("1-2-3"), &rank_size)); in TEST_F()
34 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("hccl_world_group"), &rank_size)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | group_manager.cc | 79 return CommManager::GetInstance().CreateGroupSync(group_name, ranks); in CreateGroupByExecutor()
92 return CommManager::GetInstance().DestroyGroup(group_name); in DestroyGroupByExecutor()
113 ret = CommManager::GetInstance().CreateGroupSync(group.first, group.second); in CreateGroups()
167 (void)CommManager::GetInstance().GetRankSize(world_group_, &world_size); in CreateGroup()
255 bool ret = CommManager::GetInstance().GetRankID(name, rank_id); in GetRankID()
268 bool ret = CommManager::GetInstance().GetRankSize(name, rank_size); in GetRankSize()
|
| D | step_parallel.cc | 2896 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in GetCommInfo()
2915 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetCommInfo()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 50 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRank()
90 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in PipelineSplit()
|
| D | pipeline.cc | 1334 bool ret = CommManager::GetInstance().GetRankID("", &rank_id); in GetHcclRankId()
1343 bool ret = CommManager::GetInstance().GetRankSize("", &rank_size); in GetHcclRankSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | executor.cc | 165 void CreateCommGroupTask::Run() { result_ = CommManager::GetInstance().CreateGroupSync(group_name_,… in Run()
167 void DestroyCommGroupTask::Run() { result_ = CommManager::GetInstance().DestroyGroup(group_name_); } in Run()
|
| D | session_basic.cc | 2744 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRankId()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 156 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in SetOpMappingInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.cc | 502 if (!CommManager::GetInstance().GetRankID(kNcclWorldGroup, &rank_id)) { in GetRankID()
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | dump_json_parser.cc | 560 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in GetOpOverflowBinPath()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 655 if (!CommManager::GetInstance().GetRankSize(group, &num_of_devices)) { in operator()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 658 if (!CommManager::GetInstance().GetRankID(kHcclWorldGroup, &rank_id)) { in GetDumpPath()
|