From 9d1d3583c01cf7303ae08785b95962b94c44a44f Mon Sep 17 00:00:00 2001
From: zhaozhixuan <zhaozhixuan2@hisilicon.com>
Date: Tue, 23 Mar 2021 10:29:20 +0800
Subject: [PATCH] Add ut.

---
 tests/ut/ge/graph/load/davinci_model_unittest.cc | 1 +
 tests/ut/ge/hybrid/ge_hybrid_unittest.cc         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc
index 457c3b2e..fa531010 100644
--- a/tests/ut/ge/graph/load/davinci_model_unittest.cc
+++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc
@@ -900,6 +900,7 @@ TEST_F(UtestDavinciModel, Sink_time_profile) {
 
 TEST_F(UtestDavinciModel, Step_info_profile) {
   ProfilingManager::Instance().prof_cb_.msprofReporterCallback = MsprofReport;
+  ProfilingManager::Instance().device_id_.emplace_back(0);
   DavinciModel model(0, nullptr);
   EXPECT_EQ(DavinciModel::ProfileStepInfo(&model, 0), SUCCESS);
 }
diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
index 075c5f52..e377ca66 100644
--- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
+++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc
@@ -265,6 +265,7 @@ TEST_F(UtestGeHybrid, hybrid_model_executor) {
 
 TEST_F(UtestGeHybrid, Step_info_profile) {
   ProfilingManager::Instance().prof_cb_.msprofReporterCallback = MsprofReport;
+  ProfilingManager::Instance().device_id_.emplace_back(0);
   auto graph = make_shared<ComputeGraph>("graph");
   auto root_model = make_shared<GeRootModel>(graph);
   HybridModel model(root_model);