aclopSetKernelArgs
Applicability
Product |
Supported |
|---|---|
Atlas 350 Accelerator Card |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
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.