tft_register_zit_upgrade_rollback_handler
接口功能
训练框架向Processor注册升级回滚回调函数。
对于MindSpeed-LLM训练框架，回调函数已经完成适配；而对于其他框架，用户需要自行确保回调函数的安全性。
接口格式
mindio_ttp.framework_ttp.tft_register_zit_upgrade_rollback_handler(func: Callable, ctx = None)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
func |
必选 |
rollback回调函数，完成数据集回滚等重置操作。 回调函数执行超时时间默认180秒，超时会导致流程执行失败，用户可通过环境变量TTP_NORMAL_ACTION_TIME_LIMIT设置超时时间。 |
回调函数，不为空，约定该回调函数无返回值，执行失败抛出异常。 |
ctx |
可选 |
回调函数上下文。 |
默认为空。 |
返回值
无返回值，出错会打印ERROR日志、抛出异常。
父主题： API接口参考