函数:set_kernel_workspace_sizes

C函数原型

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

Python函数

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

函数功能

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

输入说明

kernel_desc:int,Kernel描述缓存,aclopKernelDesc类型的指针。

num_workspaces:int,Workspaces个数。

workspace_sizes:int,Workspaces大小的数组地址。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

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

注意事项