Skip to content

GitLab

Commit cbe9a4cb authored by Ziheng Qin's avatar Ziheng Qin Committed by Frank Lee
Browse files

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

parent 912eb58e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_layers/test_3d/test_3d.py
View file @ cbe9a4cb
...@@ -38,6 +38,7 @@ def check_layer(): ...@@ -38,6 +38,7 @@ def check_layer():
check_loss() check_loss()
check_vocab_parallel_loss() check_vocab_parallel_loss()
def check_layer_and_operation(rank, world_size, port): def check_layer_and_operation(rank, world_size, port):
disable_existing_loggers() disable_existing_loggers()
launch(config=CONFIG, rank=rank, world_size=world_size, host='localhost', port=port, backend='nccl') launch(config=CONFIG, rank=rank, world_size=world_size, host='localhost', port=port, backend='nccl')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment