Add ut.

4 years ago · e8eeace382
parent d47d7b378a
commit e8eeace382
1 changed files with 4 additions and 0 deletions
--- a/tests/ut/ge/graph/load/davinci_model_unittest.cc
+++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc
@ -988,6 +988,10 @@ TEST_F(UtestDavinciModel, NnExecute) {
  EXPECT_EQ(outputs.size(), 1);
  input_data.blobs = output_data.blobs;
  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);
 }
 }  // namespace ge