aclmdlRIGetTasksByStream
须知:本接口为试验特性,后续版本可能会存在变更,不支持应用于商用产品中。
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
☓ |
√ |
|
√ |
|
☓ |
|
☓ |
|
☓ |
功能说明
获取指定Stream中的所有任务。
调用本接口之前,先调用aclmdlRIGetStreams接口获取模型运行实例关联的所有Stream,再根据指定Stream获取其中的所有任务。
通过两次调用本接口可以获取指定Stream中的所有任务:
- 第一次调用aclmdlRIGetTasksByStream接口:tasks参数处传入空指针,numTasks传入任意非负整数,调用aclmdlRIGetTasksByStream接口获取指定Stream中的所有任务数量numTasks。
- 第二次调用aclmdlRIGetTasksByStream接口:为tasks申请数组空间,大小为前一步中获取的numTasks,再将tasks、numTasks作为输入传入aclmdlRIGetTasksByStream接口,获取指定Stream中的所有任务。
函数原型
1 | aclError aclmdlRIGetTasksByStream(aclrtStream stream, aclmdlRITask *tasks, uint32_t *numTasks) |
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
streams |
输入 |
指定Stream。类型定义请参见aclrtStream。 需传入与模型运行实例有关联的Stream。 |
tasks |
输入&输出 |
若tasks传入空指针,表示获取指定Stream中的所有任务数量,该数量通过numTasks参数返回。 若tasks是一个数组,其大小为numTasks,则表示用于存放指定Stream中的所有任务。类型定义请参见aclmdlRITask。 |
numTasks |
输入&输出 |
作为输入时,表示tasks数组大小。 作为输出时,表示指定Stream中的所有任务数量。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。
父主题: 模型运行实例管理