开发者
资源

ttp_init_processor

接口功能

初始化MindIO TTP Processor模块。

接口格式

mindio_ttp.ttp_init_processor(rank, dp_ranks, dp_size, world_size)

接口参数

参数

是否必选

说明

取值要求

rank

必选

当前执行训练任务NPU卡号。

真实rank id。

dp_ranks

必选

本进程rank所在的DP Group对应的所有rank列表。

有效rank列表。

dp_size

必选

dp_ranks包含的rank个数。

与dp_ranks一致。

world_size

必选

参与训练任务的集群卡数。

真实有效的NPU个数。

返回值

  • 0:成功
  • 1:失败