add retore info api cn

This commit is contained in:
yao_yf 2022-03-15 17:35:01 +08:00
parent a25c5b46aa
commit cd5d1b8293
3 changed files with 22 additions and 2 deletions

View File

@ -87,6 +87,7 @@ mindspore
mindspore.merge_sliced_parameter
mindspore.load_distributed_checkpoint
mindspore.async_ckpt_thread_status
mindspore.restore_group_info_list
即时编译
--------

View File

@ -0,0 +1,19 @@
mindspore.restore_group_info_list
=======================================
.. py:class:: mindspore.restore_group_info_list(group_info_file_name)
从group_info_file_name指向的文件中提取得到通信域的信息在该通信域内的所有设备的checkpoint文件均与存储group_info_file_name的设备相同可以直接进行替换。通过配置环境变量GROUP_INFO_FILE以在编译阶段存储下该通信域信息例如"export GROUP_INFO_FILE=/data/group_info.pb"。
**参数:**
**group_info_file_name** (str) - 保存通信域的文件的名字。
**返回:**
List通信域列表。
**异常:**
- **ValueError** - 通信域文件格式不正确。
- **TypeError** - `group_info_file_name` 不是字符串。

View File

@ -491,7 +491,7 @@ Status MatMulBase::GenerateStrategiesNotPower2(int64_t stage_id, size_t dev_num_
}
}
if (strategy_cost_.empty()) {
MS_LOG(EXCEPTION) << "No available strategy.";
MS_LOG(EXCEPTION) << name_ << " : No available strategy.";
}
return SUCCESS;
}
@ -530,7 +530,7 @@ Status MatMulBase::GenerateStrategies(int64_t stage_id) {
}
}
if (strategy_cost_.empty()) {
MS_LOG(EXCEPTION) << "No available strategy.";
MS_LOG(EXCEPTION) << name_ << " : No available strategy.";
}
return SUCCESS;
}