aclopSetKernelWorkspaceSizes

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

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 address of the array 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 address of the array of workspace sizes.

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

Parent topic: Single-Operator Model Execution