a946079208
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!19641 save group info before create group 

Merge pull request !19641 from yangzhenzhang/group-info-ckpt-r1.3
This commit is contained in:
i-robot 2021-07-08 06:51:39 +00:00 committed by Gitee
parent e372634d16 b6003e6043
commit e5dcdbd366
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mindspore/ccsrc/frontend/parallel/group_manager.cc
View File

@ -190,15 +190,16 @@ Status GroupManager::CreateGroup(const std::string &group_name, const std::vecto
MS_EXCEPTION_IF_NULL(context_ptr);
std::string device_name = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET);
uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID);
bool ret = CreateGroupByExecutor(device_name, group_name, ranks, device_id);
if (!ret) {
MS_LOG(ERROR) << "Create group failed, group name is " << group_name;
return Status::FAILED;
}
std::pair<std::string, std::vector<uint32_t>> group_info = std::make_pair(group_name, ranks);
group_info_.push_back(group_info);
bool ret = CreateGroupByExecutor(device_name, group_name, ranks, device_id);
if (!ret) {
MS_LOG(WARNING) << "Create group failed, group name is " << group_name;
return Status::FAILED;
}
MS_LOG(INFO) << "Create group success, group name is " << group_name;
return Status::SUCCESS;
}