aclrtSnapShotCallbackRegister

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

注册一个回调函数,该回调函数将在快照操作的不同阶段被调用。不支持重复注册。

函数原型

aclError aclrtSnapShotCallbackRegister(aclrtSnapShotStage stage, aclrtSnapShotCallBack callback, void *args)

参数说明

参数名 输入/输出 说明
stage 输入 指定触发回调的快照阶段。
callback 输入 指向回调函数的指针。当指定的快照阶段到达时,系统将自动调用此函数。
函数定义如下:
typedef uint32_t (*aclrtSnapShotCallBack)(int32_t deviceId, void* args);
args 输入 用户自定义参数指针,在回调函数调用时传递,可以为NULL,表示不需要传递额外参数。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError