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