aclopSetKernelWorkspaceSizes

函数功能

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

函数原型

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

参数说明

参数名

输入/输出

说明

kernelDesc

输入

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

typedef struct aclopKernelDesc aclopKernelDesc

numWorkspaces

输入

Workspaces个数。

workspaceSizes

输入

Workspaces大小的数组地址指针。

返回值说明

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