aclopSetKernelArgs

Applicable Products

Product

Supported

Ascend 950PR/Ascend 950DT

×

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

√

Function Usage

Sets tiling arguments and number of concurrency times of a dynamic-shape operator.

Prototype

1
2
3
4
5
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 concurrent AI Cores for 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 of the tiling arguments, in bytes

Returns

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