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