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 |
必选 |
DP Size大小。 |
|
enable_local_copy |
必选 |
是否启用local copy。 |
|
返回值
- 0:成功
- 1:失败
父主题: API接口参考