DSPLIB_matMul_fixed_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_fixed_priv.h
 *
 *  @brief    File to hold private parameter of function DSPLIB_matMul_fixed
 *
 *  @version  0.1 - Mar 2022 : Initial Version
 *
 ********************************************************************************/
 
#ifndef DSPLIB_MATMUL_FIXED_IXX_IXX_OXX_PRIV_H_
#define DSPLIB_MATMUL_FIXED_IXX_IXX_OXX_PRIV_H_
 
#include "../common/DSPLIB_utility.h"
#include "DSPLIB_matMul_fixed.h"
 
#define DSPLIB_MATMUL_FIXED_IXX_IXX_OXX_PBLOCK_SIZE (3 * SE_PARAM_SIZE)
#define DSPLIB_MATMAPY_FXD_I8S_O8S int8_t, c7x::int_vec
#define DSPLIB_MATMAPY_FXD_I16S_O16S int16_t, c7x::long_vec
/* #define DSPLIB_matMul_fixed_SE_UNROLL_FACTOR (8) */
 
typedef DSPLIB_STATUS (*pFxnDSPLIB_matMul_fixed_exec)(DSPLIB_kernelHandle handle,
                                                      void *restrict pIn1,
                                                      void *restrict pIn2,
                                                      void *restrict pOut);
 
template <typename dataTypeIn, typename dataTypeOut>
DSPLIB_STATUS DSPLIB_matMul_fixed_init_ci(DSPLIB_kernelHandle                 handle,
                                          const DSPLIB_bufParams2D_t         *bufParamsIn0,
                                          const DSPLIB_bufParams2D_t         *bufParamsIn1,
                                          const DSPLIB_bufParams2D_t         *bufParamsOut,
                                          const DSPLIB_matMul_fixed_InitArgs *pKerInitArgs);
 
template <typename dataTypeIn, typename dataTypeOut>
DSPLIB_STATUS
DSPLIB_matMul_fixed_exec_ci(DSPLIB_kernelHandle handle, void *restrict pIn1, void *restrict pIn2, void *restrict pOut);
 
template <typename dataType>
extern DSPLIB_STATUS
DSPLIB_matMul_fixed_exec_cn(DSPLIB_kernelHandle handle, void *restrict pIn1, void *restrict pIn2, void *restrict pOut);
 
typedef struct {
   pFxnDSPLIB_matMul_fixed_exec execute;
   int32_t                      M;
   int32_t                      N;
   int32_t                      K;
   int32_t                      qs;
   int32_t                      strideIn0Elements;
   int32_t                      strideIn1Elements;
   int32_t                      strideOutElements;
   int32_t                      MBlocks;
   int32_t                      KBlocks;
   int32_t                      NBlocks;
   uint8_t                      bufPblock[DSPLIB_MATMUL_FIXED_IXX_IXX_OXX_PBLOCK_SIZE];
 
} DSPLIB_matMul_fixed_PrivArgs;
 
DSPLIB_STATUS
DSPLIB_matMul_fixed_init_cn(DSPLIB_kernelHandle   handle,
                            DSPLIB_bufParams2D_t *bufParamsIn0,
                            DSPLIB_bufParams2D_t *bufParamsIn1,
                            DSPLIB_bufParams2D_t *bufParamsOut);
 
template <typename dataTypeIn, typename dataTypeOut>
DSPLIB_STATUS DSPLIB_matMul_fixed_generic_init_ci(DSPLIB_kernelHandle                 handle,
                                                  const DSPLIB_bufParams2D_t         *bufParamsIn0,
                                                  const DSPLIB_bufParams2D_t         *bufParamsIn1,
                                                  const DSPLIB_bufParams2D_t         *bufParamsOut,
                                                  const DSPLIB_matMul_fixed_InitArgs *pKerInitArgs);
 
template <typename dataTypeIn, typename dataTypeOut>
DSPLIB_STATUS DSPLIB_matMul_fixed_generic_exec_ci(DSPLIB_kernelHandle handle,
                                                  void *restrict pIn0,
                                                  void *restrict pIn1,
                                                  void *restrict pOut);
 
template <typename dataType> void printStridedMat(char *inStr, int Nrows, int colStride, dataType *inMat)
{
   printf("Nrwo: %d, colStride: %d\n", Nrows, colStride);
   printf("%s:\n", inStr);
   dataType *printMat = inMat;
   for (int i = 0; i < Nrows; i++) {
      for (int j = 0; j < colStride; j++) {
         printf("%8d", (int16_t) *printMat++);
      }
      printf("\n");
   }
   printf("\n");
}
template void printStridedMat<int16_t>(char *inStr, int Nrows, int colStride, int16_t *inMat);
template void printStridedMat<int8_t>(char *inStr, int Nrows, int colStride, int8_t *inMat);
// template void printStridedMat<float>(char *inStr, int Nrows, int colStride, float *inMat);
// template void printStridedMat<double>(char *inStr, int Nrows, int colStride, double *inMat);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_fixed_generic_init_ci(DSPLIB_kernelHandle                 handle,
                                                  const DSPLIB_bufParams2D_t         *bufParamsIn0,
                                                  const DSPLIB_bufParams2D_t         *bufParamsIn1,
                                                  const DSPLIB_bufParams2D_t         *bufParamsOut,
                                                  const DSPLIB_matMul_fixed_InitArgs *pKerInitArgs);
 
template <typename dataTypeIn, typename dataTypeOut>
DSPLIB_STATUS DSPLIB_matMul_fixed_generic_exec_ci(DSPLIB_kernelHandle handle,
                                                  void *restrict pIn0,
                                                  void *restrict pIn1,
                                                  void *restrict pOut);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_fixed_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_fixed_InitArgs *pKerInitArgs);
 
template <typename dataType>
DSPLIB_STATUS DSPLIB_matMul_fixed_N_unroll_exec_ci(DSPLIB_kernelHandle handle,
                                                   void *restrict pIn0,
                                                   void *restrict pIn1,
                                                   void *restrict pOut);
 
#endif /* DSPLIB_MATMUL_FIXED_IXX_IXX_OXX_PRIV_H_ */
 
/* ======================================================================== */
/*  End of file:  DSPLIB_matMul_fixed.h          */
/* ======================================================================== */