RMUL2025/lib/cmsis_5/CMSIS/DSP/Testing/Source/Tests/UnaryTestsF16.cpp

617 lines
20 KiB
C++
Executable File

#include "UnaryTestsF16.h"
#include <stdio.h>
#include "Error.h"
#define SNR_THRESHOLD 57
/*
Reference patterns are generated with
a double precision computation.
*/
#define REL_ERROR (1.1e-3)
#define ABS_ERROR (1.1e-3)
/*
Comparisons for inverse
*/
/* Not very accurate for big matrix.
But big matrix needed for checking the vectorized code */
#define SNR_THRESHOLD_INV 45
#define REL_ERROR_INV (3.0e-2)
#define ABS_ERROR_INV (3.0e-2)
#define REL_ERROR_SOLVE (6.0e-2)
#define ABS_ERROR_SOLVE (2.0e-2)
/*
Comparison for Cholesky
*/
#define SNR_THRESHOLD_CHOL 45
#define REL_ERROR_CHOL (3.0e-3)
#define ABS_ERROR_CHOL (3.0e-2)
/* Upper bound of maximum matrix dimension used by Python */
#define MAXMATRIXDIM 40
#define LOADDATA2() \
const float16_t *inp1=input1.ptr(); \
const float16_t *inp2=input2.ptr(); \
\
float16_t *ap=a.ptr(); \
float16_t *bp=b.ptr(); \
\
float16_t *outp=output.ptr(); \
int16_t *dimsp = dims.ptr(); \
int nbMatrixes = dims.nbSamples() >> 1;\
int rows,columns; \
int i;
#define LOADDATA1() \
const float16_t *inp1=input1.ptr(); \
\
float16_t *ap=a.ptr(); \
\
float16_t *outp=output.ptr(); \
int16_t *dimsp = dims.ptr(); \
int nbMatrixes = dims.nbSamples() >> 1;\
int rows,columns; \
int i;
#define PREPAREDATA2() \
in1.numRows=rows; \
in1.numCols=columns; \
memcpy((void*)ap,(const void*)inp1,sizeof(float16_t)*rows*columns);\
in1.pData = ap; \
\
in2.numRows=rows; \
in2.numCols=columns; \
memcpy((void*)bp,(const void*)inp2,sizeof(float16_t)*rows*columns);\
in2.pData = bp; \
\
out.numRows=rows; \
out.numCols=columns; \
out.pData = outp;
#define PREPAREDATALT() \
in1.numRows=rows; \
in1.numCols=rows; \
memcpy((void*)ap,(const void*)inp1,sizeof(float16_t)*rows*rows); \
in1.pData = ap; \
\
in2.numRows=rows; \
in2.numCols=columns; \
memcpy((void*)bp,(const void*)inp2,sizeof(float16_t)*rows*columns);\
in2.pData = bp; \
\
out.numRows=rows; \
out.numCols=columns; \
out.pData = outp;
#define PREPAREDATA1(TRANSPOSED) \
in1.numRows=rows; \
in1.numCols=columns; \
memcpy((void*)ap,(const void*)inp1,sizeof(float16_t)*rows*columns);\
in1.pData = ap; \
\
if (TRANSPOSED) \
{ \
out.numRows=columns; \
out.numCols=rows; \
} \
else \
{ \
out.numRows=rows; \
out.numCols=columns; \
} \
out.pData = outp;
#define PREPAREDATA1C(TRANSPOSED) \
in1.numRows=rows; \
in1.numCols=columns; \
memcpy((void*)ap,(const void*)inp1,2*sizeof(float16_t)*rows*columns);\
in1.pData = ap; \
\
if (TRANSPOSED) \
{ \
out.numRows=columns; \
out.numCols=rows; \
} \
else \
{ \
out.numRows=rows; \
out.numCols=columns; \
} \
out.pData = outp;
#define LOADVECDATA2() \
const float16_t *inp1=input1.ptr(); \
const float16_t *inp2=input2.ptr(); \
\
float16_t *ap=a.ptr(); \
float16_t *bp=b.ptr(); \
\
float16_t *outp=output.ptr(); \
int16_t *dimsp = dims.ptr(); \
int nbMatrixes = dims.nbSamples() / 2;\
int rows,internal; \
int i;
#define PREPAREVECDATA2() \
in1.numRows=rows; \
in1.numCols=internal; \
memcpy((void*)ap,(const void*)inp1,sizeof(float16_t)*rows*internal);\
in1.pData = ap; \
\
memcpy((void*)bp,(const void*)inp2,sizeof(float16_t)*internal);
void UnaryTestsF16::test_mat_vec_mult_f16()
{
LOADVECDATA2();
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
internal = *dimsp++;
PREPAREVECDATA2();
arm_mat_vec_mult_f16(&this->in1, bp, outp);
outp += rows ;
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
void UnaryTestsF16::test_mat_add_f16()
{
LOADDATA2();
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATA2();
status=arm_mat_add_f16(&this->in1,&this->in2,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
void UnaryTestsF16::test_mat_sub_f16()
{
LOADDATA2();
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATA2();
status=arm_mat_sub_f16(&this->in1,&this->in2,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
void UnaryTestsF16::test_mat_scale_f16()
{
LOADDATA1();
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATA1(false);
status=arm_mat_scale_f16(&this->in1,0.5f,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
void UnaryTestsF16::test_mat_trans_f16()
{
LOADDATA1();
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATA1(true);
status=arm_mat_trans_f16(&this->in1,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
void UnaryTestsF16::test_mat_cmplx_trans_f16()
{
LOADDATA1();
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATA1C(true);
status=arm_mat_cmplx_trans_f16(&this->in1,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += 2*(rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR,REL_ERROR);
}
static void refInnerTail(float16_t *b)
{
b[0] = 1.0f;
b[1] = -2.0f;
b[2] = 3.0f;
b[3] = -4.0f;
}
static void checkInnerTail(float16_t *b)
{
ASSERT_TRUE(b[0] == 1.0f);
ASSERT_TRUE(b[1] == -2.0f);
ASSERT_TRUE(b[2] == 3.0f);
ASSERT_TRUE(b[3] == -4.0f);
}
void UnaryTestsF16::test_mat_inverse_f16()
{
const float16_t *inp1=input1.ptr();
float16_t *ap=a.ptr();
float16_t *outp=output.ptr();
int16_t *dimsp = dims.ptr();
int nbMatrixes = dims.nbSamples();
int rows,columns;
int i;
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = rows;
PREPAREDATA1(false);
refInnerTail(outp+(rows * columns));
status=arm_mat_inverse_f16(&this->in1,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
inp1 += (rows * columns);
checkInnerTail(outp);
}
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD_INV);
ASSERT_CLOSE_ERROR(output,ref,ABS_ERROR_INV,REL_ERROR_INV);
}
void UnaryTestsF16::test_mat_cholesky_dpo_f16()
{
float16_t *ap=a.ptr();
const float16_t *inp1=input1.ptr();
float16_t *outp=output.ptr();
int16_t *dimsp = dims.ptr();
int nbMatrixes = dims.nbSamples();
int rows,columns;
int i;
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = rows;
PREPAREDATA1(false);
status=arm_mat_cholesky_f16(&this->in1,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
inp1 += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD_CHOL);
ASSERT_CLOSE_ERROR(ref,output,ABS_ERROR_CHOL,REL_ERROR_CHOL);
}
void UnaryTestsF16::test_solve_upper_triangular_f16()
{
float16_t *ap=a.ptr();
const float16_t *inp1=input1.ptr();
float16_t *bp=b.ptr();
const float16_t *inp2=input2.ptr();
float16_t *outp=output.ptr();
int16_t *dimsp = dims.ptr();
int nbMatrixes = dims.nbSamples() >> 1;
int rows,columns;
int i;
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATALT();
status=arm_mat_solve_upper_triangular_f16(&this->in1,&this->in2,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
inp1 += (rows * rows);
inp2 += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(ref,output,ABS_ERROR_SOLVE,REL_ERROR_SOLVE);
}
void UnaryTestsF16::test_solve_lower_triangular_f16()
{
float16_t *ap=a.ptr();
const float16_t *inp1=input1.ptr();
float16_t *bp=b.ptr();
const float16_t *inp2=input2.ptr();
float16_t *outp=output.ptr();
int16_t *dimsp = dims.ptr();
int nbMatrixes = dims.nbSamples()>>1;
int rows,columns;
int i;
arm_status status;
for(i=0;i < nbMatrixes ; i ++)
{
rows = *dimsp++;
columns = *dimsp++;
PREPAREDATALT();
status=arm_mat_solve_lower_triangular_f16(&this->in1,&this->in2,&this->out);
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
outp += (rows * columns);
inp1 += (rows * rows);
inp2 += (rows * columns);
}
ASSERT_EMPTY_TAIL(output);
ASSERT_SNR(output,ref,(float16_t)SNR_THRESHOLD);
ASSERT_CLOSE_ERROR(ref,output,ABS_ERROR_SOLVE,REL_ERROR_SOLVE);
}
void UnaryTestsF16::setUp(Testing::testID_t id,std::vector<Testing::param_t>& params,Client::PatternMgr *mgr)
{
(void)params;
switch(id)
{
case TEST_MAT_ADD_F16_1:
input1.reload(UnaryTestsF16::INPUTS1_F16_ID,mgr);
input2.reload(UnaryTestsF16::INPUTS2_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFADD1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPB_F16_ID,mgr);
break;
case TEST_MAT_SUB_F16_2:
input1.reload(UnaryTestsF16::INPUTS1_F16_ID,mgr);
input2.reload(UnaryTestsF16::INPUTS2_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFSUB1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPB_F16_ID,mgr);
break;
case TEST_MAT_SCALE_F16_3:
input1.reload(UnaryTestsF16::INPUTS1_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFSCALE1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
break;
case TEST_MAT_TRANS_F16_4:
input1.reload(UnaryTestsF16::INPUTS1_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFTRANS1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
break;
case TEST_MAT_INVERSE_F16_5:
input1.reload(UnaryTestsF16::INPUTSINV_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSINVERT1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFINV1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
break;
case TEST_MAT_VEC_MULT_F16_6:
input1.reload(UnaryTestsF16::INPUTS1_F16_ID,mgr);
input2.reload(UnaryTestsF16::INPUTVEC1_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFVECMUL1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
b.create(MAXMATRIXDIM,UnaryTestsF16::TMPB_F16_ID,mgr);
break;
case TEST_MAT_CMPLX_TRANS_F16_7:
input1.reload(UnaryTestsF16::INPUTSC1_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSUNARY1_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFTRANSC1_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
break;
case TEST_MAT_CHOLESKY_DPO_F16_8:
input1.reload(UnaryTestsF16::INPUTSCHOLESKY1_DPO_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIMSCHOLESKY1_DPO_S16_ID,mgr);
ref.reload(UnaryTestsF16::REFCHOLESKY1_DPO_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
break;
case TEST_SOLVE_UPPER_TRIANGULAR_F16_9:
input1.reload(UnaryTestsF16::INPUT_MAT_UTSOLVE_F16_ID,mgr);
input2.reload(UnaryTestsF16::INPUT_VEC_LTSOLVE_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIM_LTSOLVE_F16_ID,mgr);
ref.reload(UnaryTestsF16::REF_UT_SOLVE_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPB_F16_ID,mgr);
break;
case TEST_SOLVE_LOWER_TRIANGULAR_F16_10:
input1.reload(UnaryTestsF16::INPUT_MAT_LTSOLVE_F16_ID,mgr);
input2.reload(UnaryTestsF16::INPUT_VEC_LTSOLVE_F16_ID,mgr);
dims.reload(UnaryTestsF16::DIM_LTSOLVE_F16_ID,mgr);
ref.reload(UnaryTestsF16::REF_LT_SOLVE_F16_ID,mgr);
output.create(ref.nbSamples(),UnaryTestsF16::OUT_F16_ID,mgr);
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPA_F16_ID,mgr);
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsF16::TMPB_F16_ID,mgr);
break;
}
}
void UnaryTestsF16::tearDown(Testing::testID_t id,Client::PatternMgr *mgr)
{
(void)id;
//output.dump(mgr);
(void)mgr;
}