ttp_init_processor
接口功能
RC2版本使用,初始化MindIO TFT Processor模块。
接口格式
mindio_ttp.ttp_init_processor(rank: int, dp_ranks: list, dp_size: int, world_size: int, replica: int, enable_local_copy: bool)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
---|---|---|---|
rank |
必选 |
当前执行训练任务NPU卡号。 |
真实rank id。 |
dp_ranks |
必选 |
当前执行训练任务NPU的DP分组的全部卡号列表。 |
真实dp_ranks的列表。 |
dp_size |
必选 |
DP大小,用于计算副本位置。 |
DP大小。 |
world_size |
必选 |
执行训练任务NPU卡数。 |
正整数。 |
replica |
必选 |
适配老版本的参数,废弃。 |
int。 |
enable_local_copy |
必选 |
表示是否开启local copy。 |
|
返回值
无
父主题: API接口参考