From 55dc62571fc862430ea76054aa2cae315a533709 Mon Sep 17 00:00:00 2001 From: chuxing Date: Sun, 28 Mar 2021 14:54:28 +0800 Subject: [PATCH] fix ut --- tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index 2166b274..8e32dd4f 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -315,6 +315,11 @@ TEST_F(UtestGeHybrid, test_parse_parallel_group) { ASSERT_EQ(builder.parallel_group_to_nodes_["group_1"].size(), 2); ASSERT_EQ(builder.parallel_group_to_nodes_["group_2"].size(), 1); + builder.parallel_group_to_nodes_.clear(); + builder.node_ref_inputs_.clear(); + model.node_items_[node] = std::move(node_item); + model.node_items_[node_1] = std::move(node_item_1); + ASSERT_FALSE(node_item->has_observer); ASSERT_TRUE(node_item_1->dependents_for_execution.empty()); ASSERT_EQ(builder.ParseDependentByParallelGroup(), SUCCESS);