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

feat(utils/gputest.py): fix lint error

pull/456/head
huangting4201 2023-10-20 16:25:08 +08:00
parent eac382ad0a
commit 2acf9b817f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/utils/gputest.py
View File

@ -45,7 +45,7 @@ def empty_cache_and_diag(batch_count, interval=50):
# # import time
# # time.sleep(10)
# print(e, "rank = ", gpc.get_global_rank(), flush=True)
# torch.cuda.memory._dump_snapshot(f"my_snapshot_{gpc.get_global_rank()}.pickle")
# torch.cuda.memory._dump_snapshot(f"my_snapshot_{gpc.get_global_rank()}.pickle")
# do empty_cache after the bench
torch.cuda.empty_cache()