昇腾社区首页
中文
注册

tft_register_stream_sync_handler

接口功能

注册同步回调函数。

对于MindSpeed-LLM和MindSpeed训练框架,回调函数已由MindIO TFT完成适配;其他框架,回调函数的安全性需由用户保证。

接口格式

mindio_ttp.framework_ttp.tft_register_stream_sync_handler(func: Callable, ctx=None)

接口参数

参数

是否必选

说明

取值要求

func

必选

同步回调函数,完成训练暂停后同步操作。避免在暂停训练后算子队列有残留算子未执行完。

回调函数,不为空。回调函数无参数,约定该回调函数无返回值,执行失败抛出异常。

ctx

-

回调函数上下文。

由注册方决定。

返回值

无返回值,出错会打印ERROR日志、抛出异常。