vm_gemm_batch_ctx struct

#include <src/features/gemm.c>

Public variables

vm_float_t*const * C

Output panels.

int ldc

Leading dimension of each C.

const vm_float_t*const * A

Input A panels (or NULL).

int lda

Leading dimension of each A.

const vm_float_t*const * B

Per-batch B panels (or NULL).

const vm_float_t* B_shared

Shared B when shared is set.

int ldb

Leading dimension of B.

int M

int N

int K

Panel sizes.

vm_float_t alpha

vm_float_t beta

GEMM scales.

bool transA

bool transB

Transpose flags.

vm_layout_t layout

Memory layout.

int shared

Non-zero when B is shared.