函数：set_kernel_workspace_sizes

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

动态Shape场景下，设置算子Workspace参数。

C函数原型

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

python函数

ret = acl.op.set_kernel_workspace_sizes(kernel_desc, num_workspaces, workspace_sizes)

参数名	说明
kernel_desc	int，Kernel描述缓存，aclopKernelDesc类型的指针地址。
num_workspaces	int，Workspaces个数。
workspace_sizes	int，Workspaces大小的数组地址。

返回值	说明
ret	int，错误码，返回0表示成功，返回其它值表示失败。

为非必须接口，根据算子情况可选。

父主题： 单算子模型执行