GeMM - 2024.1 English

Vitis Libraries

Release Date
2024-05-30
Version
2024.1 English

Matrix Multiply/GEMM (General Matrix Multiply) solution.

// classes

template <
    typename TT_DATA_A,
    typename TT_DATA_B,
    unsigned int TP_DIM_A,
    unsigned int TP_DIM_AB,
    unsigned int TP_DIM_B,
    unsigned int TP_SHIFT,
    unsigned int TP_RND,
    unsigned int TP_DIM_A_LEADING = ROW_MAJOR,
    unsigned int TP_DIM_B_LEADING = COL_MAJOR,
    unsigned int TP_DIM_OUT_LEADING = ROW_MAJOR,
    unsigned int TP_ADD_TILING_A = 1,
    unsigned int TP_ADD_TILING_B = 1,
    unsigned int TP_ADD_DETILING_OUT = 1,
    unsigned int TP_INPUT_WINDOW_VSIZE_A = TP_DIM_A* TP_DIM_AB,
    unsigned int TP_INPUT_WINDOW_VSIZE_B = TP_DIM_B* TP_DIM_AB,
    unsigned int TP_CASC_LEN = 1,
    unsigned int TP_SAT = 1,
    unsigned int TP_SSR = 1
    >
class xf::dsp::aie::blas::matrix_mult::matrix_mult_graph