add retore info api cn
This commit is contained in:
parent
a25c5b46aa
commit
cd5d1b8293
|
@ -87,6 +87,7 @@ mindspore
|
|||
mindspore.merge_sliced_parameter
|
||||
mindspore.load_distributed_checkpoint
|
||||
mindspore.async_ckpt_thread_status
|
||||
mindspore.restore_group_info_list
|
||||
|
||||
即时编译
|
||||
--------
|
||||
|
|
|
@ -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` 不是字符串。
|
|
@ -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;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue