Browse Source

ut

tags/v1.3.0
wjm 3 years ago
parent
commit
805999046a
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      tests/ut/ge/hybrid/ge_hybrid_unittest.cc

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

@@ -240,7 +240,8 @@ TEST_F(UtestGeHybrid, index_taskdefs_success) {
GeRootModelPtr ge_root_model = make_shared<GeRootModel>(graph);
HybridModel hybrid_model(ge_root_model);
HybridModelBuilder hybrid_model_builder(hybrid_model);

hybrid_model_builder.build();
hybrid_model_builder.BuildForSingleOp();
ASSERT_EQ(hybrid_model_builder.IndexTaskDefs(graph, ge_model), SUCCESS);
}

@@ -655,4 +656,4 @@ TEST_F(UtestGeHybrid, TestParseDependentInputNodesForHccl) {
ASSERT_EQ(builder.ParseDependentInputNodes(*node_item_1, deps), SUCCESS);
ASSERT_TRUE(model.GetNodeItem(node)->has_observer);
ASSERT_EQ(node_item_1->dependents_for_execution.size(), 1);
}
}

Loading…
Cancel
Save