github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-04-28 11:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish tests/test_layers/test_3d/test_3d.py code style (#1740)

Browse Source
This commit is contained in:
Ziheng Qin 2022-10-18 19:32:19 -07:00 committed by Frank Lee
parent 912eb58ea0
commit cbe9a4cb45
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_layers/test_3d/test_3d.py
View File

@ -38,6 +38,7 @@ def check_layer():
check_loss()
check_vocab_parallel_loss()
def check_layer_and_operation(rank, world_size, port):
disable_existing_loggers()
launch(config=CONFIG, rank=rank, world_size=world_size, host='localhost', port=port, backend='nccl')