From 805999046ae4e76ade663f658406ba7aaa52e942 Mon Sep 17 00:00:00 2001 From: wjm Date: Sun, 28 Feb 2021 18:24:46 +0800 Subject: [PATCH] ut --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index 25115340..6fa17861 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -240,7 +240,8 @@ TEST_F(UtestGeHybrid, index_taskdefs_success) { GeRootModelPtr ge_root_model = make_shared(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); -} \ No newline at end of file +}