aclrtSubscribeHostFunc
Atlas 200/300/500 推理产品,不支持该接口。
Atlas 推理系列产品,不支持该接口。
Atlas 200/500 A2推理产品,不支持该接口。
Atlas 训练系列产品,不支持该接口。
Atlas A2训练系列产品/Atlas 800I A2推理产品,不支持该接口。
函数功能
模型中有CPU算子、且调用aclmdlExecuteV2或aclmdlExecuteAsyncV2接口执行模型推理时,调用本接口注册处理Stream上回调函数的线程(线程需由用户自行创建),再配合调用aclrtProcessHostFunc接口触发回调函数、在模型执行之后调用aclrtUnSubscribeHostFunc接口取消注册。
约束说明
- 支持多次调用aclrtSubscribeHostFunc接口给多个Stream(仅支持同一Device内的多个Stream)注册同一个处理回调函数的线程。
- 为确保Stream内的任务按调用顺序执行,不支持调用aclrtSubscribeHostFunc接口给同一个Stream注册多个处理回调函数的线程。
函数原型
aclError aclrtSubscribeHostFunc(uint64_t hostFuncThreadId, aclrtStream exeStream)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
hostFuncThreadId |
输入 |
指定线程的ID。 |
exeStream |
输入 |
指定Stream。 |
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: 执行控制