mirror of
https://github.com/Fishwaldo/bl_mcu_sdk.git
synced 2025-07-23 21:29:17 +00:00
[feat][nmsis] add nmsis component and nn,dsp demo
This commit is contained in:
parent
b2aada479b
commit
5d1126d0f0
989 changed files with 286224 additions and 0 deletions
54
examples/dsp/MatrixFunctions_q15Part/mat_scale.c
Normal file
54
examples/dsp/MatrixFunctions_q15Part/mat_scale.c
Normal 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;
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue