123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143 |
- #include "arm_math.h"
- void arm_dot_prod_q31(
- q31_t * pSrcA,
- q31_t * pSrcB,
- uint32_t blockSize,
- q63_t * result)
- {
- q63_t sum = 0;
- uint32_t blkCnt;
- #ifndef ARM_MATH_CM0_FAMILY
- q31_t inA1, inA2, inA3, inA4;
- q31_t inB1, inB2, inB3, inB4;
-
- blkCnt = blockSize >> 2u;
-
- while(blkCnt > 0u)
- {
-
-
- inA1 = *pSrcA++;
- inA2 = *pSrcA++;
- inA3 = *pSrcA++;
- inA4 = *pSrcA++;
- inB1 = *pSrcB++;
- inB2 = *pSrcB++;
- inB3 = *pSrcB++;
- inB4 = *pSrcB++;
- sum += ((q63_t) inA1 * inB1) >> 14u;
- sum += ((q63_t) inA2 * inB2) >> 14u;
- sum += ((q63_t) inA3 * inB3) >> 14u;
- sum += ((q63_t) inA4 * inB4) >> 14u;
-
- blkCnt--;
- }
-
- blkCnt = blockSize % 0x4u;
- #else
-
-
- blkCnt = blockSize;
- #endif
- while(blkCnt > 0u)
- {
-
-
- sum += ((q63_t) * pSrcA++ * *pSrcB++) >> 14u;
-
- blkCnt--;
- }
-
- *result = sum;
- }
|