msptiEnableCallback

产品支持情况

产品

是否支持

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

x

Atlas 训练系列产品

x

Atlas 200/300/500 推理产品

x

功能说明

为特定domain和CallbackId的订阅者开启或关闭回调。

当这个CallbackId所在位置触发时,MSPTI会主动调用msptiSubscribe接口注册的回调函数。

函数原型

1
msptiResult msptiEnableCallback(uint32_t enable, msptiSubscriberHandle subscriber,                            msptiCallbackDomain domain, msptiCallbackId cbid)

参数说明

表1 参数说明

参数名

输入/输出

说明

enable

输入

回调的开关,配置该参数表示开启,未配置表示关闭。

subscriber

输入

订阅者句柄。

domain

输入

组件,当前仅支持Runtime。

cbid

输入

回调ID,标识Domain的调用点。可调用msptiCallbackIdRuntimemsptiCallbackIdHccl

返回值说明

返回MSPTI_SUCCESS表示成功,用户、域或* cbid无效时返回MSPTI_ERROR_INVALID_PARAMETER,表示失败。