aclopSetKernelWorkspaceSizes

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	√

动态Shape场景下，设置算子Workspace参数。为非必须接口，根据算子情况可选。

aclError aclopSetKernelWorkspaceSizes(aclopKernelDesc *kernelDesc, int numWorkspaces, size_t *workspaceSizes)

参数名	输入/输出	说明
kernelDesc	输入	Kernel描述缓存，aclopKernelDesc类型的指针。 typedef struct aclopKernelDesc aclopKernelDesc
numWorkspaces	输入	Workspaces个数。
workspaceSizes	输入	Workspaces大小的数组地址指针。

参数名

输入/输出

说明

kernelDesc

输入

Kernel描述缓存，aclopKernelDesc类型的指针。

typedef struct aclopKernelDesc aclopKernelDesc

numWorkspaces

输入

Workspaces个数。

workspaceSizes

输入

Workspaces大小的数组地址指针。

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

父主题： 单算子模型执行