aclopSetKernelWorkspaceSizes

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Sets the workspace parameters of an operator in dynamic shape scenarios. This API is optional. Use it as required.

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

Parameter	Input/Output	Description
kernelDesc	Input	Kernel description buffer, which is a pointer to the aclopKernelDesc type typedef struct aclopKernelDesc aclopKernelDesc;
numWorkspaces	Input	Number of workspaces
workspaceSizes	Input	Pointer to the array address of workspace sizes

Parameter

Input/Output

Description

kernelDesc

Input

Kernel description buffer, which is a pointer to the aclopKernelDesc type

typedef struct aclopKernelDesc aclopKernelDesc;

numWorkspaces

Input

Number of workspaces

workspaceSizes

Input

Pointer to the array address of workspace sizes

0 on success; else, failure. For details, see aclError.

Parent topic: Single-Operator Model Execution