aclopSetKernelWorkspaceSizes
函数功能
动态Shape场景下,设置算子Workspace参数。为非必须接口,根据算子情况可选。
函数原型
aclError aclopSetKernelWorkspaceSizes(aclopKernelDesc *kernelDesc, int numWorkspaces, size_t *workspaceSizes)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
kernelDesc |
输入 |
Kernel描述缓存,aclopKernelDesc类型的指针。 typedef struct aclopKernelDesc aclopKernelDesc |
numWorkspaces |
输入 |
Workspaces个数。 |
workspaceSizes |
输入 |
Workspaces大小的数组地址指针。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: 单算子模型执行