tft_register_repair_handler

func

必选

repair回调函数，完成优化器修复等数据修复功能。

回调函数执行超时时间默认180秒，超时会导致流程执行失败，用户可通过设置环境变量TTP_NORMAL_ACTION_TIME_LIMIT设置超时时间。

回调函数，不为空，回调函数入参要求请参见表1，约定该回调函数无返回值，执行失败抛出异常。

ctx

可选

回调上下文。

默认为空。

step

-

修复时对应的step。

正整数。

need_rebuild

-

修复是否需要重建模型和优化器。

• False：无需重建。
• True：需要重建。

error_ranks

-

需要修复的故障卡list。

list。

repair_info

-

修复策略dict，其中优化器类型按照ATTENTION：0、MOE：1的关系对应。

{
"type": int，优化器类型
"repair_type": Enum，枚举类型取值参考5.34-RepairType
"src": list，优化器修复数据的来源卡列表
"dst": list，优化器修复数据的目的卡列表
"rank_list": list，修复通信组建立所需要的卡列表
}

args

-

tft_set_step_args设置的参数。

由注册方决定。

ctx

-

回调函数上下文。

由注册方决定。