ut

tests/ut/ge/hybrid/ge_hybrid_unittest.cc

@@ -240,7 +240,8 @@ TEST_F(UtestGeHybrid, index_taskdefs_success) {
   GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph);
   HybridModel hybrid_model(ge_root_model);
   HybridModelBuilder hybrid_model_builder(hybrid_model);
-
+  hybrid_model_builder.build();
+  hybrid_model_builder.BuildForSingleOp();
   ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), SUCCESS);
 }
 
@@ -655,4 +656,4 @@ TEST_F(UtestGeHybrid, TestParseDependentInputNodesForHccl) {
   ASSERT_EQ(builder.ParseDependentInputNodes(*node_item_1, deps), SUCCESS);
   ASSERT_TRUE(model.GetNodeItem(node)->has_observer);
   ASSERT_EQ(node_item_1->dependents_for_execution.size(), 1);
-}
+}