You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.5 KiB
C++
95 lines
2.5 KiB
C++
#include "SupportTestsF32.h"
|
|
#include "Error.h"
|
|
#include "arm_math.h"
|
|
#include "Test.h"
|
|
|
|
#include <cstdio>
|
|
|
|
|
|
void SupportTestsF32::test_barycenter_f32()
|
|
{
|
|
const float32_t *inp = input.ptr();
|
|
const float32_t *coefsp = coefs.ptr();
|
|
|
|
float32_t *outp = output.ptr();
|
|
|
|
for(int i=0; i < this->nbPatterns ; i ++)
|
|
{
|
|
arm_barycenter_f32(inp, coefsp,
|
|
outp,
|
|
this->nbVectors,
|
|
this->vecDim);
|
|
|
|
inp += this->vecDim * this->nbVectors;
|
|
coefsp += this->nbVectors;
|
|
outp += this->vecDim;
|
|
}
|
|
|
|
ASSERT_NEAR_EQ(output,ref,(float32_t)1e-3);
|
|
}
|
|
|
|
void SupportTestsF32::test_weighted_sum_f32()
|
|
{
|
|
const float32_t *inp = input.ptr();
|
|
const float32_t *coefsp = coefs.ptr();
|
|
|
|
float32_t *outp = output.ptr();
|
|
|
|
for(int i=0; i < this->nbPatterns ; i ++)
|
|
{
|
|
*outp=arm_weighted_sum_f32(inp, coefsp,
|
|
this->vecDim);
|
|
|
|
inp += this->vecDim;
|
|
coefsp += this->vecDim;
|
|
outp++;
|
|
}
|
|
|
|
ASSERT_NEAR_EQ(output,ref,(float32_t)1e-3);
|
|
}
|
|
|
|
|
|
void SupportTestsF32::setUp(Testing::testID_t id,std::vector<Testing::param_t>& paramsArgs,Client::PatternMgr *mgr)
|
|
{
|
|
|
|
switch(id)
|
|
{
|
|
case TEST_BARYCENTER_F32_1:
|
|
{
|
|
input.reload(SupportTestsF32::INPUTS1_F32_ID,mgr);
|
|
coefs.reload(SupportTestsF32::WEIGHTS1_F32_ID,mgr);
|
|
dims.reload(SupportTestsF32::DIMS1_S16_ID,mgr);
|
|
ref.reload(SupportTestsF32::REF1_F32_ID,mgr);
|
|
|
|
const int16_t *dimsp = dims.ptr();
|
|
|
|
this->nbPatterns=dimsp[0];
|
|
this->nbVectors=dimsp[1];
|
|
this->vecDim=dimsp[2];
|
|
output.create(this->nbPatterns*this->vecDim,SupportTestsF32::OUT_F32_ID,mgr);
|
|
}
|
|
break;
|
|
|
|
case TEST_WEIGHTED_SUM_F32_2:
|
|
input.reload(SupportTestsF32::INPUTS2_F32_ID,mgr);
|
|
coefs.reload(SupportTestsF32::WEIGHTS2_F32_ID,mgr);
|
|
dims.reload(SupportTestsF32::DIMS2_S16_ID,mgr);
|
|
ref.reload(SupportTestsF32::REF2_F32_ID,mgr);
|
|
|
|
const int16_t *dimsp = dims.ptr();
|
|
|
|
this->nbPatterns=dimsp[0];
|
|
this->vecDim=dimsp[1];
|
|
output.create(this->nbPatterns,SupportTestsF32::OUT_F32_ID,mgr);
|
|
break;
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
void SupportTestsF32::tearDown(Testing::testID_t id,Client::PatternMgr *mgr)
|
|
{
|
|
output.dump(mgr);
|
|
}
|