昇腾社区首页
中文
注册

函数:set_exception_info_callback

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

设置异常回调函数。

函数原型

  • C函数原型
    1
    aclError aclrtSetExceptionInfoCallback(aclrtExceptionInfoCallback callback)
    
  • python函数
    1
    ret = acl.rt.set_exception_info_callback(fn)
    

参数说明

参数名

说明

callback

function,表示Python侧的回调函数,格式如下:

def exception_callback(exception_info)
"""
:exception_info: 表示异常信息aclrtExceptionInfo的指针地址
:return:
"""

返回值说明

返回值

说明

ret

int,错误码,返回0表示成功,返回其它值表示失败。

约束说明

  • 您需要在执行异步任务之前,设置异常回调函数,当Device上的任务执行异常时,系统会向用户设置的异常回调函数中传入一个包含任务ID、Stream ID、线程ID、Device ID以及错误码的aclrtExceptionInfo的指针地址,并执行回调函数,用户可以再分别调用acl.rt.get_task_id_from_exception_infoacl.rt.get_stream_id_from_exception_infoacl.rt.get_thread_id_from_exception_infoacl.rt.get_device_id_from_exception_infoacl.rt.get_error_code_from_exception_info接口获取产生异常的任务ID、Stream ID、线程ID、Device ID以及错误码,便于定位问题。

    使用场景举例:例如,在调用acl.op.execute_v2接口前,调用acl.rt.set_exception_info_callback接口设置异常回调函数,当算子在Device执行异常时,系统会向用户设置的异常回调函数中传入一个包含任务ID、Stream ID、线程ID、Device ID以及错误码的aclrtExceptionInfo的指针地址,并执行回调函数。

  • 如果多次设置异常回调函数,以最后一次设置为准。
  • 如果想清空回调函数,可再次调用acl.rt.set_exception_info_callback接口进行重置(传入None或不传入参数)。

资源参考

接口调用示例,参见AI Core异常信息获取