diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index 0325a7de..c95ab8c2 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -663,8 +663,8 @@ target_link_libraries(ge_runner PRIVATE ge_memory adump_server static_mmpa - -Wl,--whole-archive msprofiler_fwk + -Wl,--whole-archive -Wl,--no-whole-archive -Wl,--no-as-needed graph diff --git a/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc b/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc index ef19b516..2e5ad5b0 100644 --- a/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc +++ b/tests/ut/ge/profiling/ge_profiling_manager_unittest.cc @@ -66,3 +66,18 @@ struct MsprofGeOptions prof_conf = {{ 0 }}; Status ret = ProfilingManager::Instance().ParseOptions(options.profiling_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); +}