colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a unit_test for div_uint8 (#251) 

Signed-off-by: Chen Xin <jack.chen@verisilicon.com>

Co-authored-by: Chen Xin <jack.chen@verisilicon.com>
This commit is contained in:
chxin66 2021-12-30 13:31:30 +08:00 committed by GitHub
parent 6275f84575
commit eecbe264b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/tim/vx/ops/elementwise_test.cc
View File

@ -235,3 +235,49 @@ TEST(Div, shape_5_1_broadcast_scale_uint8) {
EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data()));
EXPECT_EQ(golden, output);
}
TEST(Div, Div_uint8) {
auto context = tim::vx::Context::Create();
auto graph = context->CreateGraph();
tim::vx::ShapeType input_shape({2, 3, 1, 1});
tim::vx::Quantization input_quant(tim::vx::QuantType::ASYMMETRIC, 1.0, 0);
tim::vx::TensorSpec input_spec(tim::vx::DataType::UINT8, input_shape,
tim::vx::TensorAttribute::INPUT, input_quant);
uint8_t data1[] = {1, 2, 3, 4, 5, 6};
uint8_t data2[] = {2, 2, 2, 2, 2, 2};
auto input1 = graph->CreateTensor(input_spec, data1);
auto input2 = graph->CreateTensor(input_spec, data2);
tim::vx::ShapeType output_shape({2, 3, 1, 1});
tim::vx::Quantization output_quant(tim::vx::QuantType::ASYMMETRIC, 1.0, 0);
tim::vx::TensorSpec output_spec(tim::vx::DataType::UINT8, output_shape,
tim::vx::TensorAttribute::OUTPUT,
output_quant);
auto output = graph->CreateTensor(output_spec);
auto op = graph->CreateOperation<tim::vx::ops::Div>();
(*op).BindInputs({input1, input2}).BindOutputs({output});
if (!graph->Compile()) {
std::cout << "Compile graph fail." << std::endl;
EXPECT_TRUE(-1);
}
graph->PrintGraph();
if (!graph->Run()) {
std::cout << "Run graph fail." << std::endl;
EXPECT_TRUE(-1);
}
std::vector<uint8_t> output_data;
std::vector<uint8_t> golden={0,1,2,2,2,3,0,0,0,0};
output_data.resize(1 * 10);
if (!output->CopyDataFromTensor(output_data.data())) {
std::cout << "Copy output data fail." << std::endl;
EXPECT_TRUE(-1);
}
EXPECT_EQ(golden, output_data);
}