diff --git a/torch_npu/profiler/analysis/prof_common_func/_constant.py b/torch_npu/profiler/analysis/prof_common_func/_constant.py index b9a0ecbe21394e6e9db78a98024f890c353152a5..9c8540c1ab356a8c4207de96c8640481d93ec560 100644 --- a/torch_npu/profiler/analysis/prof_common_func/_constant.py +++ b/torch_npu/profiler/analysis/prof_common_func/_constant.py @@ -104,6 +104,7 @@ class Constant(object): RANK_ID = "rank_id" COMMON_CONFIG = "common_config" ACTIVITIES = "activities" + CPU_ACTIVITIES = "ProfilerActivity.CPU" NPU_ACTIVITIES = "ProfilerActivity.NPU" EXPERIMENTAL_CONFIG = "experimental_config" PROFILER_LEVEL = '_profiler_level'