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