fix ut

4 years ago · 55dc62571f
parent a89298a045
commit 55dc62571f
1 changed files with 5 additions and 0 deletions
--- 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);