arm_mat_mult_q31.c

Go to the documentation of this file.

/* ----------------------------------------------------------------------      
* Copyright (C) 2011 ARM Limited. All rights reserved. 
*      
* $Date:        15. December 2011   
* $Revision:    V2.0.0  
*      
* Project:      Cortex-R DSP Library 
* Title:        arm_mat_mult_q31.c      
*      
* Description:   Q31 matrix multiplication.      
*      
* Target Processor:          Cortex-R4/R5
*
* Version 1.0.0 2011/03/08
*     Alpha release.
*
* Version 1.0.1 2011/09/30
*     Beta release.
*
* Version 2.0.0 2011/12/15
*     Final release. 
* 
* -------------------------------------------------------------------- */     
     
#include "arm_math.h"     
     
arm_status arm_mat_mult_q31(     
  const arm_matrix_instance_q31 * pSrcA,     
  const arm_matrix_instance_q31 * pSrcB,     
  arm_matrix_instance_q31 * pDst)     
{     
  q31_t *pIn1 = pSrcA->pData;                    /* input data matrix pointer A */     
  q31_t *pIn2 = pSrcB->pData;                    /* input data matrix pointer B */     
  q31_t *pInA = pSrcA->pData;                    /* input data matrix pointer A */     
  q31_t *pOut = pDst->pData;                     /* output data matrix pointer */     
  q31_t *px;                                     /* Temporary output data matrix pointer */     
  q63_t sum;                                     /* Accumulator */     
  uint16_t numRowsA = pSrcA->numRows;            /* number of rows of input matrix A    */     
  uint16_t numColsB = pSrcB->numCols;            /* number of columns of input matrix B */     
  uint16_t numColsA = pSrcA->numCols;            /* number of columns of input matrix A */     
  uint16_t col, i = 0u, j, row = numRowsA, colCnt;      /* loop counters */     
  arm_status status;                             /* status of matrix multiplication */     
  q31_t a0, a1, a2, a3, b0, b1, b2, b3;   
     
#ifdef ARM_MATH_MATRIX_CHECK     
  /* Check for matrix mismatch condition */     
  if((pSrcA->numCols != pSrcB->numRows) ||     
     (pSrcA->numRows != pDst->numRows) || (pSrcB->numCols != pDst->numCols))     
  {     
    /* Set status as ARM_MATH_SIZE_MISMATCH */     
    status = ARM_MATH_SIZE_MISMATCH;     
  }     
  else     
#endif     
  {     
    /* The following loop performs the dot-product of each row in pSrcA with each column in pSrcB */     
    /* row loop */     
    do     
    {     
      /* Output pointer is set to starting address of the row being processed */     
      px = pOut + i;     
     
      /* For every row wise process, the column loop counter is to be initiated */     
      col = numColsB;     
     
      /* For every row wise process, the pIn2 pointer is set      
       ** to the starting address of the pSrcB data */     
      pIn2 = pSrcB->pData;     
     
      j = 0u;     
     
      /* column loop */     
      do     
      {     
        /* Set the variable sum, that acts as accumulator, to zero */     
        sum = 0;     
     
        /* Initiate the pointer pIn1 to point to the starting address of pInA */     
        pIn1 = pInA;     
     
        /* Apply loop unrolling and compute 4 MACs simultaneously. */     
        colCnt = numColsA >> 2;     
     
     
        /* matrix multiplication */     
        while(colCnt > 0u)     
        {     
          /* c(m,n) = a(1,1)*b(1,1) + a(1,2) * b(2,1) + .... + a(m,p)*b(p,n) */     
          /* Perform the multiply-accumulates */      
          b0 = *pIn2;   
          pIn2 += numColsB;   
   
          a0 = *pIn1++;   
          a1 = *pIn1++;   
   
          b1 = *pIn2;   
          pIn2 += numColsB;   
          b2 = *pIn2;   
          pIn2 += numColsB;   
   
          sum += (q63_t) a0 * b0;   
          sum += (q63_t) a1 * b1;   
   
          a2 = *pIn1++;   
          a3 = *pIn1++;   
   
          b3 = *pIn2;   
          pIn2 += numColsB;   
             
          sum += (q63_t) a2 * b2;   
          sum += (q63_t) a3 * b3;   
   
          /* Decrement the loop counter */     
          colCnt--;     
        }     
     
        /* If the columns of pSrcA is not a multiple of 4, compute any remaining output samples here.      
         ** No loop unrolling is used. */     
        colCnt = numColsA % 0x4u;     
     
        while(colCnt > 0u)     
        {     
          /* c(m,n) = a(1,1)*b(1,1) + a(1,2) * b(2,1) + .... + a(m,p)*b(p,n) */     
          /* Perform the multiply-accumulates */     
          a0 = * pIn1++;   
          sum += (q63_t) a0 * *pIn2;     
          pIn2 += numColsB;     
     
          /* Decrement the loop counter */     
          colCnt--;     
        }     
     
        /* Convert the result from 2.62 to 1.31 format and store in destination buffer */     
        *px++ = (q31_t) (sum >> 31);     
     
        /* Update the pointer pIn2 to point to the  starting address of the next column */     
        j++;     
        pIn2 = (pSrcB->pData) + j;     
     
        /* Decrement the column loop counter */     
        col--;     
     
      } while(col > 0u);     
     
      /* Update the pointer pInA to point to the  starting address of the next row */     
      i = i + numColsB;     
      pInA = pInA + numColsA;     
     
      /* Decrement the row loop counter */     
      row--;     
     
    } while(row > 0u);     
     
    /* set status as ARM_MATH_SUCCESS */     
    status = ARM_MATH_SUCCESS;     
  }     
  /* Return to application */     
  return (status);     
}