[feat][nmsis] add nmsis component and nn,dsp demo

This commit is contained in:
jzlv 2021-09-26 13:38:51 +08:00
parent b2aada479b
commit 5d1126d0f0
989 changed files with 286224 additions and 0 deletions

View file

@ -0,0 +1,54 @@
#include "ref.h"
riscv_status ref_mat_scale_f32(const riscv_matrix_instance_f32 *pSrc,
float32_t scale, riscv_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)pSrc->numRows * pSrc->numCols;
for (i = 0; i < numSamples; i++) {
pDst->pData[i] = pSrc->pData[i] * scale;
}
return RISCV_MATH_SUCCESS;
}
riscv_status ref_mat_scale_q31(const riscv_matrix_instance_q31 *pSrc, q31_t scale,
int32_t shift, riscv_matrix_instance_q31 *pDst)
{
uint32_t i;
uint32_t numSamples; /* total number of elements in the matrix */
int32_t totShift = shift + 1;
q31_t tmp;
/* Total number of samples in the input matrix */
numSamples = (uint32_t)pSrc->numRows * pSrc->numCols;
for (i = 0; i < numSamples; i++) {
tmp = ((q63_t)pSrc->pData[i] * scale) >> 32;
pDst->pData[i] = ref_sat_q31((q63_t)tmp << totShift);
}
return RISCV_MATH_SUCCESS;
}
riscv_status ref_mat_scale_q15(const riscv_matrix_instance_q15 *pSrc, q15_t scale,
int32_t shift, riscv_matrix_instance_q15 *pDst)
{
uint32_t i;
uint32_t numSamples; /* total number of elements in the matrix */
int32_t totShift = 15 - shift;
/* Total number of samples in the input matrix */
numSamples = (uint32_t)pSrc->numRows * pSrc->numCols;
for (i = 0; i < numSamples; i++) {
pDst->pData[i] =
ref_sat_q15(((q31_t)pSrc->pData[i] * scale) >> totShift);
}
return RISCV_MATH_SUCCESS;
}