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

Minor update to graph_test 

Update graph_test to demonstrate that CompileToBinary
does not require input data.

Signed-off-by: Kainan Cha <kainan.zha@verisilicon.com>
This commit is contained in:
Kainan Cha 2021-05-12 13:31:19 +08:00
parent 301d88a5a6
commit 52401b52a5
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/tim/vx/graph_test.cc
View File

@ -39,7 +39,7 @@ TEST(graph, gen_binary_graph_with_empty_graph) {
} }
TEST(graph, gen_binary_graph_with_simple_add) { TEST(graph, gen_binary_graph_with_simple_add) {
auto ctx = tim::vx::Context::Create(); auto ctx = tim::vx::Context::Create();
auto graph = ctx->CreateGraph(); auto graph = ctx->CreateGraph();
tim::vx::ShapeType io_shape({1,1,1,1}); tim::vx::ShapeType io_shape({1,1,1,1});
@ -47,13 +47,7 @@ TEST(graph, gen_binary_graph_with_simple_add) {
tim::vx::TensorSpec output_spec(tim::vx::DataType::FLOAT32, io_shape, tim::vx::TensorAttribute::OUTPUT); tim::vx::TensorSpec output_spec(tim::vx::DataType::FLOAT32, io_shape, tim::vx::TensorAttribute::OUTPUT);
auto input_t0 = graph->CreateTensor(input_spec); auto input_t0 = graph->CreateTensor(input_spec);
auto input_t1 = graph->CreateTensor(input_spec); auto input_t1 = graph->CreateTensor(input_spec);
auto output_t = graph->CreateTensor(output_spec); auto output_t = graph->CreateTensor(output_spec);
float in = 1.0f;
float expected_out = 2.0f;
EXPECT_TRUE(input_t0->CopyDataToTensor(&in, sizeof(in)));
EXPECT_TRUE(input_t1->CopyDataToTensor(&in, sizeof(in)));
auto add = graph->CreateOperation<tim::vx::ops::Add>(); auto add = graph->CreateOperation<tim::vx::ops::Add>();
(*add).BindInputs({input_t0, input_t1}).BindOutputs({output_t}); (*add).BindInputs({input_t0, input_t1}).BindOutputs({output_t});
@ -63,9 +57,15 @@ TEST(graph, gen_binary_graph_with_simple_add) {
EXPECT_NE(bin_size, -1); EXPECT_NE(bin_size, -1);
std::vector<char> nbg_buf(bin_size); std::vector<char> nbg_buf(bin_size);
// generate binary graph does't require input data
EXPECT_TRUE(graph->CompileToBinary(nbg_buf.data(), &bin_size)); EXPECT_TRUE(graph->CompileToBinary(nbg_buf.data(), &bin_size));
// binary graph compilation doesn't impact current graph's execution // binary graph compilation doesn't impact current graph's execution
float in = 1.0f;
float expected_out = 2.0f;
EXPECT_TRUE(input_t0->CopyDataToTensor(&in, sizeof(in)));
EXPECT_TRUE(input_t1->CopyDataToTensor(&in, sizeof(in)));
EXPECT_TRUE(graph->Run()); EXPECT_TRUE(graph->Run());
float output = 0.0f; float output = 0.0f;
EXPECT_TRUE(output_t->CopyDataFromTensor(&output)); EXPECT_TRUE(output_t->CopyDataFromTensor(&output));
@ -75,7 +75,7 @@ TEST(graph, gen_binary_graph_with_simple_add) {
auto nbg_in0 = nbg_graph->CreateTensor(input_spec); auto nbg_in0 = nbg_graph->CreateTensor(input_spec);
auto nbg_in1 = nbg_graph->CreateTensor(input_spec); auto nbg_in1 = nbg_graph->CreateTensor(input_spec);
auto nbg_out = nbg_graph->CreateTensor(output_spec); auto nbg_out = nbg_graph->CreateTensor(output_spec);
EXPECT_TRUE(nbg_in0->CopyDataToTensor(&in, sizeof(in))); EXPECT_TRUE(nbg_in0->CopyDataToTensor(&in, sizeof(in)));
EXPECT_TRUE(nbg_in1->CopyDataToTensor(&in, sizeof(in))); EXPECT_TRUE(nbg_in1->CopyDataToTensor(&in, sizeof(in)));
@ -89,4 +89,4 @@ TEST(graph, gen_binary_graph_with_simple_add) {
output=0.0f; output=0.0f;
EXPECT_TRUE(nbg_out->CopyDataFromTensor(&output)); EXPECT_TRUE(nbg_out->CopyDataFromTensor(&output));
EXPECT_EQ(output, expected_out); EXPECT_EQ(output, expected_out);
} }