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

!18145 Fixed the problem that the value of source card or destination card is not intuitive 

Merge pull request !18145 from 张毅辉/cluster_profiler_for_slow_net
This commit is contained in:
i-robot 2021-06-11 11:09:44 +08:00 committed by Gitee
parent b159ca939f ef438dc752
commit 7d76feec4f
13 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/space_to_depth_base.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/exp_fp32.c Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/space_to_batch_fp32.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/gather_parameter.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gatherNd_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/gather_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/leaky_relu_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/quant_dtype_cast_int8.c Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/sigmoid_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/squeeze_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/unsqueeze_int8.h Executable file → Normal file
View File

0
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/space_to_depth_parameter.h Executable file → Normal file
View File

13
mindspore/profiler/parser/hccl_parser.py
View File

@ -200,9 +200,11 @@ class HcclParser:
info_set = set()
for item in communication_info:
# index0:step_num,index1:communication_cost,index2:communication_wait_cost,index3:link_info
info_set.add(item[0])
if item[0].isdigit():
info_set.add(int(item[0]))
info_set = sorted(info_set)
for item in info_set:
item = str(item)
step_communication_info = [info for info in communication_info if info[0] == item]
step_communication_cost = sum([i[1] for i in step_communication_info])
step_communication_wait_cost = sum([i[2] for i in step_communication_info])
@ -250,6 +252,15 @@ class HcclParser:
dst_rank = item.get("args").get("dst rank")
if src_rank is None or dst_rank is None:
continue
# When the SDMA operation is in the card,
# the source card or destination card is 0xffffffff, and it needs to be converted to localrank.
if int(src_rank) == int('0xffffffff', 16):
src_rank = dst_rank
if int(dst_rank) == int('0xffffffff', 16):
dst_rank = src_rank
if item.get("args").get("transport type") == CommunicationInfo.LOCAL.value:
item["args"]["src rank"] = dst_rank
item["args"]["dst rank"] = src_rank