Browse Source

support unknown while subgraph

pull/1381/head
lichun 4 years ago
parent
commit
0ebfac471d
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      tests/ut/ge/graph/load/davinci_model_unittest.cc

+ 4
- 4
tests/ut/ge/graph/load/davinci_model_unittest.cc View File

@@ -49,7 +49,7 @@ int32_t MsprofReport(uint32_t moduleId, uint32_t type, void *data, uint32_t len)
return 0;
}

/*TEST_F(UtestDavinciModel, init_success) {
TEST_F(UtestDavinciModel, init_success) {
DavinciModel model(0, nullptr);
ComputeGraphPtr graph = make_shared<ComputeGraph>("default");
ProfilingManager::Instance().is_load_profiling_ = true;
@@ -139,7 +139,7 @@ int32_t MsprofReport(uint32_t moduleId, uint32_t type, void *data, uint32_t len)
EXPECT_EQ(outputs.size(), 1);

ProfilingManager::Instance().is_load_profiling_ = false;
}*/
}

TEST_F(UtestDavinciModel, init_data_op) {
DavinciModel model(0, nullptr);
@@ -776,7 +776,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_input_dims_normal) {
}

// test label_set_task Init
/*TEST_F(UtestDavinciModel, label_task_success) {
TEST_F(UtestDavinciModel, label_task_success) {
DavinciModel model(0, nullptr);
ComputeGraphPtr graph = make_shared<ComputeGraph>("default");

@@ -860,7 +860,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_input_dims_normal) {
EXPECT_EQ(model.input_addrs_list_.size(), 0);
EXPECT_EQ(model.output_addrs_list_.size(), 0);
EXPECT_EQ(model.task_list_.size(), 5);
}*/
}

TEST_F(UtestDavinciModel, LoadWithQueue_fail_with_diff_args) {
DavinciModel model(0, nullptr);


Loading…
Cancel
Save