GDMA Functions

okk_gdma_32bit_cpy_S2L

void okk_gdma_32bit_cpy_S2L(local_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from system memory to local memory for 32-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_32bit_cpy_L2S

void okk_gdma_32bit_cpy_L2S(system_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from local memory to system memory for 32-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_32bit_cpy_L2L

void okk_gdma_32bit_cpy_L2L(local_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from local memory to local memory for 32-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_32bit_cpy_S2S

void okk_gdma_32bit_cpy_S2S(system_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from system memory to system memory for 32-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_32bit_matrix_S2L

void okk_gdma_32bit_matrix_S2L(local_addr_t dst_addr, system_addr_t src_addr, int rows, int cols, int cols_per_channel, int row_stride)

Copy matrix from system memory to local memory for 32-bit data type.

\[dst(x, y) = src(x, y)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in system memory.

  • rows – Number of the rows of the matrix.

  • cols – Number of the columns of the matrix.

  • cols_per_channel – Number of the columns per channel of the destination matrix tensor.

  • row_stride – Stride of the row of the source matrix tensor.

Remarks

  • The destination tensor is in the Matrix Layout.

  • The elements of each row of the source matrix are continuous.

  • The data type of the destination and source tensors is 32-bit.

okk_gdma_32bit_matrix_L2S

void okk_gdma_32bit_matrix_L2S(system_addr_t dst_addr, local_addr_t src_addr, int rows, int cols, int cols_per_channel, int row_stride)

Copy matrix from local memory to system memory for 32-bit data type.

\[dst(x, y) = src(x, y)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in local memory.

  • rows – Number of the rows of the matrix.

  • cols – Number of the columns of the matrix.

  • cols_per_channel – Number of the columns per channel of the source matrix tensor.

  • row_stride – Stride of the row of the destination matrix tensor.

Remarks

  • The elements of each row of the destination matrix are continuous.

  • The source tensor is in the Matrix Layout.

  • The data type of the destination and source tensors is 32-bit.

okk_gdma_32bit_set_C_system

void okk_gdma_32bit_set_C_system(system_addr_t dst_addr, x32 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in system memory to be a constant value for 32-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • C – Constant value of 32-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks

okk_gdma_32bit_set_C_local

void okk_gdma_32bit_set_C_local(local_addr_t dst_addr, x32 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in local memory to be a constant value for 32-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • C – Constant value of 32-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks

okk_gdma_16bit_cpy_S2L

void okk_gdma_16bit_cpy_S2L(local_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from system memory to local memory for 16-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_16bit_cpy_L2S

void okk_gdma_16bit_cpy_L2S(system_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from local memory to system memory for 16-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_16bit_cpy_L2L

void okk_gdma_16bit_cpy_L2L(local_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from local memory to local memory for 16-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_16bit_cpy_S2S

void okk_gdma_16bit_cpy_S2S(system_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from system memory to system memory for 16-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_16bit_set_C_system

void okk_gdma_16bit_set_C_system(system_addr_t dst_addr, x16 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in system memory to be a constant value for 16-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • C – Constant value of 16-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks

okk_gdma_16bit_set_C_local

void okk_gdma_16bit_set_C_local(local_addr_t dst_addr, x16 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in local memory to be a constant value for 16-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • C – Constant value of 16-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks

okk_gdma_8bit_cpy_S2L

void okk_gdma_8bit_cpy_S2L(local_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor to the destination tensor from system memory to local memory for 8-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_8bit_cpy_L2S

void okk_gdma_8bit_cpy_L2S(system_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from local memory to system memory for 8-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_8bit_cpy_L2L

void okk_gdma_8bit_cpy_L2L(local_addr_t dst_addr, local_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy the elements of the source tensor to the destination tensor from local memory to local memory for 8-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • src_addr – Address of the source tensor in local memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_8bit_cpy_S2S

void okk_gdma_8bit_cpy_S2S(system_addr_t dst_addr, system_addr_t src_addr, const dim4 *shape, const dim4 *dst_stride, const dim4 *src_stride)

Copy tensor from system memory to system memory for 8-bit data type.

\[dst(n, c, h, w) = src(n, c, h, w)\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • src_addr – Address of the source tensor in system memory.

  • shape – Pointer to the shape of the destination and source tensors.

  • dst_stride – Pointer to the stride of the destination tensor.

  • src_stride – Pointer to the stride of the source tensor.

Remarks

okk_gdma_8bit_set_C_system

void okk_gdma_8bit_set_C_system(system_addr_t dst_addr, x8 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in system memory to be a constant value for 8-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in system memory.

  • C – Constant value of 8-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks

okk_gdma_8bit_set_C_local

void okk_gdma_8bit_set_C_local(local_addr_t dst_addr, x8 C, const dim4 *shape, const dim4 *dst_stride)

Set all the elements of the destination tensor in local memory to be a constant value for 8-bit data type.

\[dst(n, c, h, w) = C\]
Parameters

  • dst_addr – Address of the destination tensor in local memory.

  • C – Constant value of 8-bit to set.

  • shape – Pointer to the shape of the destination tensor.

  • dst_stride – Pointer to the stride of the destination tensor.

Remarks