文档
注册

ttp init controller

接口功能

初始化MindIO TTP controller模块。

接口格式

mindio_ttp.ttp_init_controller(rank, world_size, replica, enable_local_copy)

接口参数

参数

是否必选

说明

取值要求

rank

必选

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

大于或等于0。

world_size

必选

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

大于0。

replica

必选

Data Parallel Size大小。

  • 偶数:len(dp_ranks)//2
  • 奇数:2

enable_local_copy

必选

是否启用local copy。

  • False
  • True(启用,适用于大集群)

返回值

  • 0:成功
  • 1:失败
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词