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,错误码。
|
约束说明 |
为非必须接口,根据算子情况可选。 |