昇腾社区首页
中文
注册

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。

  • true:开启
  • false:关闭

返回值