aclopSetKernelArgs

Applicability

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

Description

Sets tiling arguments and number of times the AI Cores will be kickstarted for concurrent execution in dynamic shape scenarios.

Prototype

aclError aclopSetKernelArgs(aclopKernelDesc *kernelDesc,
const char *kernelId,
uint32_t blockDim,
const void *args,
uint32_t argSize)

Parameters

Parameter

Input/Output

Description

kernelDesc

Input

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

typedef struct aclopKernelDesc aclopKernelDesc;

kernelId

Input

Pointer to the kernel ID, which must be the same as that passed to the aclopCreateKernel call.

blockDim

Input

Number of times the AI Cores will be kickstarted for concurrent kernel execution.

Keep the setting consistent with the number of AI Cores used in TIK operator implementation.

args

Input

Pointer to the tiling arguments.

argSize

Input

Memory size in bytes of the tiling arguments.

Returns

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