GitHub-Mirror/bl_mcu_sdk
2
0
Fork 0
mirror of https://github.com/Fishwaldo/bl_mcu_sdk.git synced 2025-07-23 05:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

61
examples/dsp/ComplexMathFunctions/cmplx_dot_prod.c Normal file
View file

@ -0,0 +1,61 @@
#include "ref.h"
void ref_cmplx_dot_prod_f32(float32_t *pSrcA, float32_t *pSrcB,
uint32_t numSamples, float32_t *realResult,
float32_t *imagResult)
{
float32_t sumr, sumi;
uint32_t i;
sumr = 0;
sumi = 0;
for (i = 0; i < numSamples * 2; i += 2) {
sumr += pSrcA[i] * pSrcB[i] - pSrcA[i + 1] * pSrcB[i + 1];
sumi += pSrcA[i] * pSrcB[i + 1] + pSrcA[i + 1] * pSrcB[i];
}
*realResult = sumr;
*imagResult = sumi;
}
void ref_cmplx_dot_prod_q31(q31_t *pSrcA, q31_t *pSrcB, uint32_t numSamples,
q63_t *realResult, q63_t *imagResult)
{
q63_t sumr, sumi;
uint32_t i;
sumr = 0;
sumi = 0;
for (i = 0; i < numSamples * 2; i += 2) {
// shifting down 14 here to provide guard bits
sumr += (((q63_t)pSrcA[i] * pSrcB[i]) >> 14) -
(((q63_t)pSrcA[i + 1] * pSrcB[i + 1]) >> 14);
sumi += (((q63_t)pSrcA[i] * pSrcB[i + 1]) >> 14) +
(((q63_t)pSrcA[i + 1] * pSrcB[i]) >> 14);
}
*realResult = sumr;
*imagResult = sumi;
}
void ref_cmplx_dot_prod_q15(q15_t *pSrcA, q15_t *pSrcB, uint32_t numSamples,
q31_t *realResult, q31_t *imagResult)
{
q63_t sumr, sumi;
uint32_t i;
sumr = 0;
sumi = 0;
for (i = 0; i < numSamples * 2; i += 2) {
sumr += (q31_t)pSrcA[i] * pSrcB[i] - (q31_t)pSrcA[i + 1] * pSrcB[i + 1];
sumi += (q31_t)pSrcA[i] * pSrcB[i + 1] + (q31_t)pSrcA[i + 1] * pSrcB[i];
}
// shifting down 6 at the end here because there are already 32 guard bits
// available, this method is more accurate
*realResult = (q31_t)(sumr >> 6);
*imagResult = (q31_t)(sumi >> 6);
}