DSPLIB_matMul_priv.h

Go to the documentation of this file.

/******************************************************************************
 *                                                                             *
 * module name       :DSPLIB                                                   *
 *                                                                             *
 * module descripton :Matrix Multiply Accelerator Library module for C7x+MMA   *
 *                                                                             *
 * Copyright (C) 2017-2018 Texas Instruments Incorporated - https://www.ti.com/ *
 * ALL RIGHTS RESERVED                                                         *
 *                                                                             *
 ******************************************************************************/
 
/********************************************************************************
 *  @file     DSPLIB_matMul_priv.h
 *
 *  @brief    File to hold private parameter of function DSPLIB_matMul
 *
 *  @version  0.1 - Mar 2022 : Initial Version
 *
 ********************************************************************************/
 
#ifndef DSPLIB_MATMUL_IXX_IXX_OXX_PRIV_H_
#define DSPLIB_MATMUL_IXX_IXX_OXX_PRIV_H_
 
#include "../common/DSPLIB_utility.h"
#include "DSPLIB_matMul.h"
 
#define DSPLIB_MATMUL_IXX_IXX_OXX_PBLOCK_SIZE (5 * SE_PARAM_SIZE)
 
/* #define DSPLIB_MATMUL_SE_UNROLL_FACTOR (8) */
 
typedef DSPLIB_STATUS (*pFxnDSPLIB_matMul_exec)(DSPLIB_kernelHandle handle,
                                                void *restrict pIn1,
                                                void *restrict pIn2,
                                                void *restrict pOut);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_init_ci(DSPLIB_kernelHandle           handle,
                                    const DSPLIB_bufParams2D_t   *bufParamsIn0,
                                    const DSPLIB_bufParams2D_t   *bufParamsIn1,
                                    const DSPLIB_bufParams2D_t   *bufParamsOut,
                                    const DSPLIB_matMul_InitArgs *pKerInitArgs);
 
template <typename dataType>
DSPLIB_STATUS
DSPLIB_matMul_exec_ci(DSPLIB_kernelHandle handle, void *restrict pIn1, void *restrict pIn2, void *restrict pOut);
 
template <typename dataType>
extern DSPLIB_STATUS
DSPLIB_matMul_exec_cn(DSPLIB_kernelHandle handle, void *restrict pIn1, void *restrict pIn2, void *restrict pOut);
 
typedef struct {
   pFxnDSPLIB_matMul_exec execute;
   int32_t                M;
   int32_t                N;
   int32_t                K;
   int32_t                strideIn0Elements;
   int32_t                strideIn1Elements;
   int32_t                strideOutElements;
   int32_t                KBlocks;
   int32_t                NBlocks;
   uint8_t                bufPblock[DSPLIB_MATMUL_IXX_IXX_OXX_PBLOCK_SIZE];
 
} DSPLIB_matMul_PrivArgs;
 
DSPLIB_STATUS
DSPLIB_matMul_init_cn(DSPLIB_kernelHandle   handle,
                      DSPLIB_bufParams2D_t *bufParamsIn0,
                      DSPLIB_bufParams2D_t *bufParamsIn1,
                      DSPLIB_bufParams2D_t *bufParamsOut);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_generic_init_ci(DSPLIB_kernelHandle           handle,
                                            const DSPLIB_bufParams2D_t   *bufParamsIn0,
                                            const DSPLIB_bufParams2D_t   *bufParamsIn1,
                                            const DSPLIB_bufParams2D_t   *bufParamsOut,
                                            const DSPLIB_matMul_InitArgs *pKerInitArgs);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_generic_exec_ci(DSPLIB_kernelHandle handle,
                                            void *restrict pIn0,
                                            void *restrict pIn1,
                                            void *restrict pOut);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_N_unroll_init_ci(DSPLIB_kernelHandle           handle,
                                             const DSPLIB_bufParams2D_t   *bufParamsIn0,
                                             const DSPLIB_bufParams2D_t   *bufParamsIn1,
                                             const DSPLIB_bufParams2D_t   *bufParamsOut,
                                             const DSPLIB_matMul_InitArgs *pKerInitArgs);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_N_unroll_exec_ci(DSPLIB_kernelHandle handle,
                                             void *restrict pIn0,
                                             void *restrict pIn1,
                                             void *restrict pOut);
 
#endif /* DSPLIB_MATMUL_IXX_IXX_OXX_PRIV_H_ */
 
/* ======================================================================== */
/*  End of file:  DSPLIB_matMul.h          */
/* ======================================================================== */