aclrtLaunchCallback

函数功能

在Stream的任务队列中增加一个需要执行的回调函数。异步接口。

函数原型

aclError aclrtLaunchCallback(aclrtCallback fn, void *userData, aclrtCallbackBlockType blockType, aclrtStream stream)

参数说明

参数名

输入/输出

说明

fn

输入

指定要增加的回调函数。

回调函数的函数原型为:

typedef void (*aclrtCallback)(void *userData)

userData

输入

待传递给回调函数的用户数据的指针。

blockType

输入

指定回调函数调用是否阻塞Device运行。

typedef enum aclrtCallbackBlockType {
    ACL_CALLBACK_NO_BLOCK,  //非阻塞,Atlas 200/300/500 推理产品不支持该选项
    ACL_CALLBACK_BLOCK,  //阻塞
} aclrtCallbackBlockType;

stream

输入

指定Stream。

返回值说明

返回0表示成功,返回其它值表示失败。

参考资源

接口调用流程及示例代码,参见异步模型推理