From 28f2a8686debde0b4ffb8d3b56e6466226943078 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Thu, 11 Mar 2021 16:08:57 +0800
Subject: [PATCH] dynamic shape inference support

---
 tests/ut/ge/generator/ge_generator_unittest.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc
index 90d209b6..9b3aadda 100644
--- a/tests/ut/ge/generator/ge_generator_unittest.cc
+++ b/tests/ut/ge/generator/ge_generator_unittest.cc
@@ -77,6 +77,7 @@ TEST_F(UtestGeGenerator, test_generate_model) {
   string file_name_prefix = "123";
   const vector<GeTensor> inputs = {};
   ModelBufferData model_data;
-  EXPECT_EQ(GenerateModel(graph, file_name_prefix, inputs, model_data), FAILED);
+  GeGenerator generator;
+  EXPECT_EQ(generator.GenerateModel(graph, file_name_prefix, inputs, model_data, true), FAILED);
 }
 }  // namespace ge