diff --git a/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc b/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc index 5027c988..70b23230 100644 --- a/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc +++ b/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc @@ -155,3 +155,18 @@ TEST_F(UtestGeProfilinganager, init_from_cfg_success1) { Status ret = ProfilingManager::Instance().Init(options); EXPECT_EQ(ret, ge::SUCCESS); } + +TEST_F(UtestGeProfilinganager, init_success2) { + setenv("PROFILING_MODE", "true", true); + Options options; + options.device_id = 0; + options.job_id = "0"; + options.profiling_mode = "1"; + options.profiling_options = R"({"result_path":"/data/profiling","training_trace":"on","task_trace":"on","aicpu_trace":"on","fp_point":"Data_0","bp_point":"addn","ai_core_metrics":"ResourceConflictRatio"})"; + string profiling_config; + + ProfilingManager::Instance().SetProfilingConfig(profiling_config); + + Status ret = ProfilingManager::Instance().Init(options); + EXPECT_EQ(ret, ge::SUCCESS); +}