Browse Source

fix root graph nullptr core dump error

pull/1410/head
lichun 4 years ago
parent
commit
951de37ae4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

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

@@ -276,9 +276,9 @@ TEST_F(UtestGeHybrid, test_parse_parallel_group) {
op_desc->SetOpKernelLibName("ops_kernel_info_hccl");
GeRootModelPtr root_model = MakeShared<ge::GeRootModel>(compute_graph);
HybridModel model(root_model);
model.root_graph_ = compute_graph;

HybridModelBuilder builder(model);
builder.root_graph_ = compute_graph;
ASSERT_EQ(builder.CollectParallelGroups(node_item.get()), SUCCESS);

ASSERT_EQ(builder.node_to_parallel_groups_.size(), 1);


Loading…
Cancel
Save