Function: kernel_args_get_mem_size

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

Obtains the actual size of the memory required by the parameter list during kernel launch.

C Prototype

aclError aclrtKernelArgsGetMemSize(aclrtFuncHandle funcHandle, size_t userArgsSize, size_t *actualArgsSize)

Python Function

actual_args_size, ret = acl.rt.kernel_args_get_mem_size(func_handle, user_args_size)

Parameter	Description
func_handle	Int, kernel function handle. Call acl.rt.binary_get_function to obtain the kernel function handle and transfer it to this API as an input parameter.
user_args_size	Int, memory size (in bytes) required for storing the parameter list data. The memory size of each parameter data must be 8-byte aligned. user_args_size is the sum of the memory sizes of the aligned parameter data.

Parameter

Description

func_handle

Int, kernel function handle.

Call acl.rt.binary_get_function to obtain the kernel function handle and transfer it to this API as an input parameter.

user_args_size

Int, memory size (in bytes) required for storing the parameter list data.

The memory size of each parameter data must be 8-byte aligned. user_args_size is the sum of the memory sizes of the aligned parameter data.

Return Value	Description
actual_args_size	Int, actual memory size (in bytes) required by the parameter list during kernel launch.
ret	Int, 0 on success; else, failure.

Parent topic: Kernel Loading and Execution