Browse Source

fix ut

tags/v1.3.0
chuxing 4 years ago
parent
commit
55dc62571f
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

+ 5
- 0
tests/ut/ge/hybrid/ge_hybrid_unittest.cc View File

@@ -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_1"].size(), 2);
ASSERT_EQ(builder.parallel_group_to_nodes_["group_2"].size(), 1); 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_FALSE(node_item->has_observer);
ASSERT_TRUE(node_item_1->dependents_for_execution.empty()); ASSERT_TRUE(node_item_1->dependents_for_execution.empty());
ASSERT_EQ(builder.ParseDependentByParallelGroup(), SUCCESS); ASSERT_EQ(builder.ParseDependentByParallelGroup(), SUCCESS);


Loading…
Cancel
Save