From e8eeace382f2b75ea87283657611d50d265cc503 Mon Sep 17 00:00:00 2001
From: zhaozhixuan <zhaozhixuan2@hisilicon.com>
Date: Sat, 27 Mar 2021 20:42:53 +0800
Subject: [PATCH] Add ut.

---
 tests/ut/ge/graph/load/davinci_model_unittest.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc
index 3487f8ed..c6f3da44 100644
--- 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