add retore info api cn
This commit is contained in:
parent
a25c5b46aa
commit
cd5d1b8293
|
@ -87,6 +87,7 @@ mindspore
|
||||||
mindspore.merge_sliced_parameter
|
mindspore.merge_sliced_parameter
|
||||||
mindspore.load_distributed_checkpoint
|
mindspore.load_distributed_checkpoint
|
||||||
mindspore.async_ckpt_thread_status
|
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()) {
|
if (strategy_cost_.empty()) {
|
||||||
MS_LOG(EXCEPTION) << "No available strategy.";
|
MS_LOG(EXCEPTION) << name_ << " : No available strategy.";
|
||||||
}
|
}
|
||||||
return SUCCESS;
|
return SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -530,7 +530,7 @@ Status MatMulBase::GenerateStrategies(int64_t stage_id) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (strategy_cost_.empty()) {
|
if (strategy_cost_.empty()) {
|
||||||
MS_LOG(EXCEPTION) << "No available strategy.";
|
MS_LOG(EXCEPTION) << name_ << " : No available strategy.";
|
||||||
}
|
}
|
||||||
return SUCCESS;
|
return SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue