diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc index a4f2f6d7..8e24919c 100644 --- a/tests/ut/ge/generator/ge_generator_unittest.cc +++ b/tests/ut/ge/generator/ge_generator_unittest.cc @@ -188,9 +188,12 @@ TEST_F(UtestGeGenerator, test_generate_online_model) { } TEST_F(UtestGeGenerator, test_check_engine_type_support) { + GeTensorDesc tensor_desc; shared_ptr op_desc = make_shared("Add", "add"); - GeGenerator generator; - generator.Initialize({}); - EXPECT_EQ(generator.CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED); + op_desc->AddInputDesc(tensor_desc); + op_desc->AddInputDesc(tensor_desc); + op_desc->AddOutputDesc(tensor_desc); + + EXPECT_EQ(CheckEngineTypeSupport(op_desc, ENGINE_AIVECTOR), FAILED); } } // namespace ge