From 645f2abfee3b7de38185ad7d7b3fbd6f4ea3f7e1 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Fri, 12 Mar 2021 11:57:37 +0800
Subject: [PATCH] dynamic shape inference support

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

diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc
index 5cf5038b..066577df 100644
--- a/tests/ut/ge/generator/ge_generator_unittest.cc
+++ b/tests/ut/ge/generator/ge_generator_unittest.cc
@@ -76,7 +76,7 @@ TEST_F(UtestGeGenerator, test_build_single_op_online) {
 TEST_F(UtestGeGenerator, test_set_model_name) {
   GeGenerator generator;
   generator.Initialize({});
-  GeRootModelPtr ge_root_model = make_shared<GeRootModelPtr>(GeRootModel());
+  GeRootModelPtr ge_root_model = make_shared<GeRootModel>(GeRootModel());
   ComputeGraphPtr graph = make_shared<ComputeGraph>(ComputeGraph("graph"));
   (void)AttrUtils::SetBool(graph, "_dynamic_shape_partitioned", true);
   ge_root_model->root_graph_ = std::move(graph);