From ce83b1569db73c2d36a3e6c3b0faa8bdf057594a Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Thu, 11 Mar 2021 11:39:08 +0800
Subject: [PATCH] 	modified:   tests/ut/ge/hybrid/ge_hybrid_unittest.cc

---
 tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 659d11c6..c6f9f4f1 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -218,6 +218,8 @@ TEST_F(UtestGeHybrid, init_weight_success) {
   graph->AddSubgraph("sub", sub_graph);
 
   GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph);
+  GeModelPtr ge_sub_model = make_shared<GeModelPtr>(sub_graph);
+  ge_root_model->SetSubgraphInstanceNameToModel("sub",ge_sub_model);
   HybridModel hybrid_model(ge_root_model);
   HybridModelBuilder hybrid_model_builder(hybrid_model);
   auto ret = hybrid_model_builder.InitWeights();