昇腾社区首页
中文
注册

register_callbacks(self, operator, func)

功能说明

向TaskD注册进程管理函数,用于后续在管理进程生命周期过程中使用。

输入参数说明

表1 输入参数说明

参数

类型

说明

operator

string

当前注入的回调类型。

  • KILL_WORKER:注册MindSpore进程的停止方法,停止特定训练进程。
  • START_ALL_WORKER:注册MindSpore进程的启动方法,启动当前节点所有的进程。
  • MONITOR:注册MindSpore进程的监测方法,返回当前本节点各rank进程信息。
  • START_WORKER_LIST:注册MindSpore进程的启动方法,启动当前节点的部分进程。

func

函数

当前注册的功能的函数回调