Function: kernel_args_get_mem_size

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

Obtains the actual memory size 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 pass it 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 all aligned parameter data.

Parameter

Description

func_handle

Int, kernel function handle.

Call acl.rt.binary_get_function to obtain the kernel function handle, and pass it 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 all 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