mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!32831 fix the bug of communication name map between hccl and step trace 

Merge pull request !32831 from zangqx/zangqx_10
This commit is contained in:
i-robot 2022-04-12 06:47:46 +00:00 committed by Gitee
parent 4183a03188 bfb834fe12
commit c6e9b9b7f9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/python/mindspore/profiler/parser/hccl_parser.py
View File

@ -198,7 +198,8 @@ class HcclParser:
communication_operator_mapping_info = dict() communication_operator_mapping_info = dict()
for hccl_key, hccl_value in op_names_in_hccl_dic.items(): for hccl_key, hccl_value in op_names_in_hccl_dic.items():
for step_trace_key, step_trace_value in op_names_in_step_trace_dic.items(): for step_trace_key, step_trace_value in op_names_in_step_trace_dic.items():
if hccl_key.lower() == step_trace_key.lower(): # the step_trace_key format is: scope/operatorName
if hccl_key.lower() == step_trace_key.lower().split('/')[-1]:
communication_operator_mapping_info[hccl_key] = list(zip(hccl_value, step_trace_value)) communication_operator_mapping_info[hccl_key] = list(zip(hccl_value, step_trace_value))
logger.info("Communication operator name mapping info is %s", communication_operator_mapping_info) logger.info("Communication operator name mapping info is %s", communication_operator_mapping_info)