p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add ACL_PROF_HCCL_TRACE Switch

This commit is contained in:
臧庆香 2021-12-06 20:24:23 +08:00
parent d8e5200e39
commit 1c82f79320
3 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphengine

@ -1 +1 @@
Subproject commit d4509b5ec9f3886e40b6c3c7c1dd59a6a0a333aa
Subproject commit 2d1c7772b4bd9715e5c73861f771529bf0109991

3
mindspore/ccsrc/profiler/device/ascend/ascend_profiling.cc
View File

@ -80,6 +80,9 @@ uint64_t AscendProfiler::GetOptionsMask() const {
if (options_json["aicpu"] == "on") {
mask |= ACL_PROF_AICPU;
}
if (options_json["hccl"] == "on") {
mask |= ACL_PROF_HCCL_TRACE;
}
return mask;
}

6
mindspore/profiler/profiling.py
View File

@ -204,6 +204,9 @@ class Profiler:
profile_memory = "off"
if self._profile_memory:
profile_memory = "on"
profiler_communication = "off"
if self._profile_communication:
profiler_communication = "on"
fp_point = os.environ.get("PROFILING_FP_START", "")
bp_point = os.environ.get("PROFILING_BP_END", "")
@ -216,7 +219,8 @@ class Profiler:
"task_trace": "on",
"aic_metrics": "ArithmeticUtilization",
"aicpu": "on",
"profile_memory": profile_memory
"profile_memory": profile_memory,
"hccl": profiler_communication
}
return profiling_options