guowenxue
2024-06-25 5fc803d51ca097f07b4efbe0290ccc540b0660df
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#include "ref.h"
 
arm_status ref_mat_add_f32(
  const arm_matrix_instance_f32 * pSrcA,
  const arm_matrix_instance_f32 * pSrcB,
  arm_matrix_instance_f32 * pDst)
{
    uint32_t i;
  uint32_t numSamples;                           /* total number of elements in the matrix  */
 
    /* Total number of samples in the input matrix */
    numSamples = (uint32_t) pSrcA->numRows * pSrcA->numCols;
    
    for(i=0;i<numSamples;i++)
    {
        pDst->pData[i] = pSrcA->pData[i] + pSrcB->pData[i];
    }
    
    return ARM_MATH_SUCCESS;
}
 
arm_status ref_mat_add_q31(
  const arm_matrix_instance_q31 * pSrcA,
  const arm_matrix_instance_q31 * pSrcB,
  arm_matrix_instance_q31 * pDst)
{
    uint32_t i;
  uint32_t numSamples;                           /* total number of elements in the matrix  */
 
    /* Total number of samples in the input matrix */
    numSamples = (uint32_t) pSrcA->numRows * pSrcA->numCols;
    
    for(i=0;i<numSamples;i++)
    {
        pDst->pData[i] = ref_sat_q31( (q63_t)pSrcA->pData[i] + pSrcB->pData[i]);
    }
    
    return ARM_MATH_SUCCESS;
}
 
arm_status ref_mat_add_q15(
  const arm_matrix_instance_q15 * pSrcA,
  const arm_matrix_instance_q15 * pSrcB,
  arm_matrix_instance_q15 * pDst)
{
    uint32_t i;
  uint32_t numSamples;                           /* total number of elements in the matrix  */
 
    /* Total number of samples in the input matrix */
    numSamples = (uint32_t) pSrcA->numRows * pSrcA->numCols;
    
    for(i=0;i<numSamples;i++)
    {
        pDst->pData[i] = ref_sat_q15( (q31_t)pSrcA->pData[i] + pSrcB->pData[i]);
    }
    
    return ARM_MATH_SUCCESS;
}