github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fixed strings in profiler outputs (#325)

pull/394/head
HELSON 2022-03-07 17:08:56 +08:00 committed by Frank Lee
parent de0468c7a8
commit 4f26fabe4f
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
colossalai/utils/profiler/comm_profiler.py
View File

@ -38,11 +38,11 @@ def _format_memory(nbytes):
MB = 1024 * KB MB = 1024 * KB
GB = 1024 * MB GB = 1024 * MB
if (abs(nbytes) >= GB): if (abs(nbytes) >= GB):
return '{:.2f} Gb'.format(nbytes * 1.0 / GB) return '{:.2f} GB'.format(nbytes * 1.0 / GB)
elif (abs(nbytes) >= MB): elif (abs(nbytes) >= MB):
return '{:.2f} Mb'.format(nbytes * 1.0 / MB) return '{:.2f} MB'.format(nbytes * 1.0 / MB)
elif (abs(nbytes) >= KB): elif (abs(nbytes) >= KB):
return '{:.2f} Kb'.format(nbytes * 1.0 / KB) return '{:.2f} KB'.format(nbytes * 1.0 / KB)
else: else:
return str(nbytes) + ' b' return str(nbytes) + ' b'
@ -52,9 +52,9 @@ def _format_bandwith(volme: float, time_us: int):
mb_per_sec = volme / time_us * sec_div_mb mb_per_sec = volme / time_us * sec_div_mb
if mb_per_sec >= 1024.0: if mb_per_sec >= 1024.0:
return '{:.3f} Gb/s'.format(mb_per_sec / 1024.0) return '{:.3f} GB/s'.format(mb_per_sec / 1024.0)
else: else:
return '{:.3f} Mb/s'.format(mb_per_sec) return '{:.3f} MB/s'.format(mb_per_sec)
class CommEvent(object): class CommEvent(object):