Browse Source

Add ut.

tags/v1.3.0
zhaozhixuan 4 years ago
parent
commit
e8eeace382
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      tests/ut/ge/graph/load/davinci_model_unittest.cc

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

@@ -988,6 +988,10 @@ TEST_F(UtestDavinciModel, NnExecute) {
EXPECT_EQ(outputs.size(), 1); EXPECT_EQ(outputs.size(), 1);
input_data.blobs = output_data.blobs; input_data.blobs = output_data.blobs;
EXPECT_EQ(input_data.blobs.size(), 1); EXPECT_EQ(input_data.blobs.size(), 1);

ProfilingManager::Instance().prof_cb_.msprofReporterCallback = MsprofReport;
ProfilingManager::Instance().device_id_.emplace_back(0);
model.task_list_.resize(1);
EXPECT_EQ(model.NnExecute(stream, false, input_data, output_data), SUCCESS); EXPECT_EQ(model.NnExecute(stream, false, input_data, output_data), SUCCESS);
} }
} // namespace ge } // namespace ge

Loading…
Cancel
Save