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日志、抛出异常。
父主题: API接口参考