|
|
|
@ -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);
|
|
|
|
|