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);