register_callbacks(self, operator, func)

Description

Registers process management functions with TaskD for future use in managing the process lifecycle.

Input Parameters

Table 1 Input parameters

Parameter

Type

Description

operator

String

Callback type to be injected.

  • KILL_WORKER: Register the method for stopping the MindSpore process to stop a specific training process.
  • START_ALL_WORKER: Register the method for starting the MindSpore process to start all processes on the current node.
  • MONITOR: Register the method for monitoring the MindSpore process to return the information about all rank processes on the current node.
  • START_WORKER_LIST: Register the method for starting the MindSpore process to start certain processes on the current node.

func

Function

Callback of the currently registered function.