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(graph); + GeModelPtr ge_sub_model = make_shared(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();