昇腾社区首页
开发者
资源
支持
积分兑换 NEW

ttp_init_controller

接口功能

初始化MindIO TTP Controller模块。

接口格式

mindio_ttp.ttp_init_controller(rank, world_size)

接口参数

参数

是否必选

说明

取值要求

rank

必选

当前执行训练任务的NPU ID。

大于或等于0。

world_size

必选

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

大于0。

返回值

  • 0：成功
  • 1：失败
父主题： API接口参考