390 lines
12 KiB
C++
Executable File
390 lines
12 KiB
C++
Executable File
#include "UnaryTestsQ15.h"
|
|
#include <stdio.h>
|
|
#include "Error.h"
|
|
|
|
#define SNR_THRESHOLD 70
|
|
|
|
/*
|
|
|
|
Reference patterns are generated with
|
|
a double precision computation.
|
|
|
|
*/
|
|
#define ABS_ERROR_Q15 ((q15_t)4)
|
|
#define ABS_ERROR_Q63 ((q63_t)(1<<16))
|
|
|
|
#define ONEHALF 0x4000
|
|
|
|
/* Upper bound of maximum matrix dimension used by Python */
|
|
#define MAXMATRIXDIM 40
|
|
|
|
static void refInnerTail(q15_t *b)
|
|
{
|
|
b[0] = 1;
|
|
b[1] = -1;
|
|
b[2] = 2;
|
|
b[3] = -2;
|
|
b[4] = 3;
|
|
b[5] = -3;
|
|
b[6] = 4;
|
|
b[7] = -4;
|
|
}
|
|
|
|
static void checkInnerTail(q15_t *b)
|
|
{
|
|
ASSERT_TRUE(b[0] == 1);
|
|
ASSERT_TRUE(b[1] == -1);
|
|
ASSERT_TRUE(b[2] == 2);
|
|
ASSERT_TRUE(b[3] == -2);
|
|
ASSERT_TRUE(b[4] == 3);
|
|
ASSERT_TRUE(b[5] == -3);
|
|
ASSERT_TRUE(b[6] == 4);
|
|
ASSERT_TRUE(b[7] == -4);
|
|
}
|
|
|
|
|
|
#define LOADDATA2() \
|
|
const q15_t *inp1=input1.ptr(); \
|
|
const q15_t *inp2=input2.ptr(); \
|
|
\
|
|
q15_t *ap=a.ptr(); \
|
|
q15_t *bp=b.ptr(); \
|
|
\
|
|
q15_t *outp=output.ptr(); \
|
|
int16_t *dimsp = dims.ptr(); \
|
|
int nbMatrixes = dims.nbSamples() >> 1;\
|
|
int rows,columns; \
|
|
int i;
|
|
|
|
#define LOADDATA1() \
|
|
const q15_t *inp1=input1.ptr(); \
|
|
\
|
|
q15_t *ap=a.ptr(); \
|
|
\
|
|
q15_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(q15_t)*rows*columns);\
|
|
in1.pData = ap; \
|
|
\
|
|
in2.numRows=rows; \
|
|
in2.numCols=columns; \
|
|
memcpy((void*)bp,(const void*)inp2,sizeof(q15_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(q15_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(q15_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 q15_t *inp1=input1.ptr(); \
|
|
const q15_t *inp2=input2.ptr(); \
|
|
\
|
|
q15_t *ap=a.ptr(); \
|
|
q15_t *bp=b.ptr(); \
|
|
\
|
|
q15_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(q15_t)*rows*internal);\
|
|
in1.pData = ap; \
|
|
\
|
|
memcpy((void*)bp,(const void*)inp2,sizeof(q15_t)*internal);
|
|
|
|
|
|
void UnaryTestsQ15::test_mat_vec_mult_q15()
|
|
{
|
|
|
|
|
|
LOADVECDATA2();
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
internal = *dimsp++;
|
|
|
|
PREPAREVECDATA2();
|
|
refInnerTail(outp + rows);
|
|
arm_mat_vec_mult_q15(&this->in1, bp, outp);
|
|
|
|
outp += rows ;
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::test_mat_add_q15()
|
|
{
|
|
LOADDATA2();
|
|
arm_status status;
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
columns = *dimsp++;
|
|
|
|
PREPAREDATA2();
|
|
refInnerTail(outp + rows * columns);
|
|
status=arm_mat_add_q15(&this->in1,&this->in2,&this->out);
|
|
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
|
|
|
|
outp += (rows * columns);
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::test_mat_sub_q15()
|
|
{
|
|
LOADDATA2();
|
|
arm_status status;
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
columns = *dimsp++;
|
|
|
|
PREPAREDATA2();
|
|
refInnerTail(outp + rows * columns);
|
|
status=arm_mat_sub_q15(&this->in1,&this->in2,&this->out);
|
|
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
|
|
|
|
outp += (rows * columns);
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::test_mat_scale_q15()
|
|
{
|
|
LOADDATA1();
|
|
arm_status status;
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
columns = *dimsp++;
|
|
|
|
PREPAREDATA1(false);
|
|
refInnerTail(outp + rows * columns);
|
|
status=arm_mat_scale_q15(&this->in1,ONEHALF,0,&this->out);
|
|
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
|
|
|
|
outp += (rows * columns);
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::test_mat_trans_q15()
|
|
{
|
|
LOADDATA1();
|
|
arm_status status;
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
columns = *dimsp++;
|
|
|
|
PREPAREDATA1(true);
|
|
refInnerTail(outp + rows * columns);
|
|
status=arm_mat_trans_q15(&this->in1,&this->out);
|
|
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
|
|
|
|
outp += (rows * columns);
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::test_mat_cmplx_trans_q15()
|
|
{
|
|
LOADDATA1();
|
|
arm_status status;
|
|
|
|
for(i=0;i < nbMatrixes ; i ++)
|
|
{
|
|
rows = *dimsp++;
|
|
columns = *dimsp++;
|
|
|
|
PREPAREDATA1C(true);
|
|
refInnerTail(outp + 2*rows * columns);
|
|
status=arm_mat_cmplx_trans_q15(&this->in1,&this->out);
|
|
ASSERT_TRUE(status==ARM_MATH_SUCCESS);
|
|
|
|
outp += 2*(rows * columns);
|
|
checkInnerTail(outp);
|
|
|
|
}
|
|
|
|
|
|
ASSERT_SNR(output,ref,(q15_t)SNR_THRESHOLD);
|
|
|
|
ASSERT_NEAR_EQ(output,ref,ABS_ERROR_Q15);
|
|
|
|
}
|
|
|
|
|
|
void UnaryTestsQ15::setUp(Testing::testID_t id,std::vector<Testing::param_t>& params,Client::PatternMgr *mgr)
|
|
{
|
|
|
|
|
|
(void)params;
|
|
switch(id)
|
|
{
|
|
case TEST_MAT_ADD_Q15_1:
|
|
input1.reload(UnaryTestsQ15::INPUTS1_Q15_ID,mgr);
|
|
input2.reload(UnaryTestsQ15::INPUTS2_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFADD1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPB_Q15_ID,mgr);
|
|
break;
|
|
|
|
case TEST_MAT_SUB_Q15_2:
|
|
input1.reload(UnaryTestsQ15::INPUTS1_Q15_ID,mgr);
|
|
input2.reload(UnaryTestsQ15::INPUTS2_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFSUB1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
b.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPB_Q15_ID,mgr);
|
|
break;
|
|
|
|
case TEST_MAT_SCALE_Q15_3:
|
|
input1.reload(UnaryTestsQ15::INPUTS1_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFSCALE1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
break;
|
|
|
|
case TEST_MAT_TRANS_Q15_4:
|
|
input1.reload(UnaryTestsQ15::INPUTS1_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFTRANS1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
break;
|
|
|
|
case TEST_MAT_VEC_MULT_Q15_5:
|
|
input1.reload(UnaryTestsQ15::INPUTS1_Q15_ID,mgr);
|
|
input2.reload(UnaryTestsQ15::INPUTVEC1_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFVECMUL1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
b.create(MAXMATRIXDIM,UnaryTestsQ15::TMPB_Q15_ID,mgr);
|
|
break;
|
|
|
|
case TEST_MAT_CMPLX_TRANS_Q15_6:
|
|
input1.reload(UnaryTestsQ15::INPUTSC1_Q15_ID,mgr);
|
|
dims.reload(UnaryTestsQ15::DIMSUNARY1_S16_ID,mgr);
|
|
|
|
ref.reload(UnaryTestsQ15::REFTRANSC1_Q15_ID,mgr);
|
|
|
|
output.create(ref.nbSamples(),UnaryTestsQ15::OUT_Q15_ID,mgr);
|
|
a.create(MAXMATRIXDIM*MAXMATRIXDIM,UnaryTestsQ15::TMPA_Q15_ID,mgr);
|
|
break;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
void UnaryTestsQ15::tearDown(Testing::testID_t id,Client::PatternMgr *mgr)
|
|
{
|
|
(void)id;
|
|
output.dump(mgr);
|
|
}
|