aclmdlExecuteAsync

产品支持情况

产品

是否支持

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

执行模型推理。

本接口是异步接口,调用接口成功仅表示任务下发成功,不表示任务执行成功。调用该接口后,需调用同步等待接口(例如,aclrtSynchronizeStream)确保任务已执行完成,否则可能会导致训练或推理等业务异常、Device断链掉卡等未知情况。

函数原型

aclError aclmdlExecuteAsync(uint32_t modelId, const aclmdlDataset *input, aclmdlDataset *output, aclrtStream stream)

参数说明

参数名

输入/输出

说明

modelId

输入

指定需要执行推理的模型的ID。

调用模型加载接口(例如aclmdlLoadFromFile接口、aclmdlLoadFromMem等)成功后,会返回模型ID,该ID作为本接口的输入。

input

输入

模型推理的输入数据的指针。

对于捕获Stream上下发任务的场景,此处传nullptr。

output

输出

模型推理的输出数据的指针。

对于捕获Stream上下发任务的场景,此处传nullptr。

stream

输入

指定Stream。

返回值说明

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

约束说明

参考资源

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