From ad108e2c3bc606909047a25664ab4b1d850d5b93 Mon Sep 17 00:00:00 2001 From: unknown Date: Wed, 3 Mar 2021 20:33:04 +0800 Subject: [PATCH] Add ut. --- tests/ut/ge/generator/ge_generator_unittest.cc | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc index 09ddf2ec..8fca22fa 100644 --- a/tests/ut/ge/generator/ge_generator_unittest.cc +++ b/tests/ut/ge/generator/ge_generator_unittest.cc @@ -53,26 +53,21 @@ TEST_F(UtestGeGenerator, test_build_single_op_offline) { EXPECT_EQ(generator.BuildSingleOpModel(op_desc, inputs, outputs, "offline_"), GE_GENERATOR_GRAPH_MANAGER_BUILD_GRAPH_FAILED); } -/* TEST_F(UtestGeGenerator, test_build_single_op_online) { - GeTensorDesc tensor_desc(GeShape(), FORMAT_NCHW, DT_FLOAT); - TensorUtils::SetSize(tensor_desc, 512); + GeTensorDesc tensor_desc; shared_ptr op_desc = make_shared("Add", "add"); - EXPECT_EQ(op_desc->AddInputDesc(tensor_desc), GRAPH_SUCCESS); - EXPECT_EQ(op_desc->AddInputDesc(tensor_desc), GRAPH_SUCCESS); - EXPECT_EQ(op_desc->AddOutputDesc(tensor_desc), GRAPH_SUCCESS); + op_desc->AddInputDesc(tensor_desc); + op_desc->AddInputDesc(tensor_desc); + op_desc->AddOutputDesc(tensor_desc); GeTensor tensor(tensor_desc); const vector inputs = { tensor, tensor }; const vector outputs = { tensor }; - // not Initialize, impl is null. GeGenerator generator; generator.Initialize({}); ModelBufferData model_buffer; - EXPECT_EQ(generator.BuildSingleOpModel(op_desc, inputs, outputs, ENGINE_SYS, model_buffer), GE_GENERATOR_GRAPH_MANAGER_BUILD_GRAPH_FAILED); + EXPECT_EQ(generator.BuildSingleOpModel(op_desc, inputs, outputs, ENGINE_AIVECTOR, model_buffer), FAILED); } -*/ - } // namespace ge