函数:execute
C函数原型 |
aclError aclmdlExecute(uint32_t modelId, const aclmdlDataset *input, aclmdlDataset *output) |
---|---|
Python函数 |
ret = acl.mdl.execute(model_id, input, output) |
函数功能 |
执行模型推理,直到返回推理结果。 |
输入说明 |
model_id:int,指定需要执行推理的模型的ID。 调用acl.mdl.load_from_file接口/acl.mdl.load_from_mem接口/acl.mdl.load_from_file_with_mem接口/acl.mdl.load_from_mem_with_mem接口加载模型成功后,会返回模型ID。 input:int,模型推理的输入数据对应的指针地址,具体请参见aclmdlDataset。 output:int,模型推理的输出数据对应的指针地址,具体请参见aclmdlDataset。 调用acl.create_data_buffer接口创建存放对应index输出数据的aclDataBuffer类型时,支持在data参数处传入0,同时size需设置为0,表示创建一个空的aclDataBuffer类型,然后在模型执行过程中,系统内部自行计算并申请该index输出的内存。使用该方式可节省内存,但内存数据使用结束后,需由用户释放内存并重置aclDataBuffer,同时,系统内部申请内存时涉及内存拷贝,可能涉及性能损耗。 释放内存并重置aclDataBuffer的示例代码如下: data_buffer = acl.mdl.get_dataset_buffer(output, 0) // 根据index获取对应的dataBuffer data_addr = acl.get_data_buffer_addr(data_buffer) // 获取data的Device指针地址 acl.rt.free(data_addr) // 释放Device内存 acl.update_data_buffer(data_buffer, 0, 0) // 重置dataBuffer,以便下次推理 |
返回值说明 |
ret:int,错误码。
|
约束说明 |
|
参考资源 |