From cbe9a4cb45dedfdbfd1f7201e26dc1ac4a93b86b Mon Sep 17 00:00:00 2001 From: Ziheng Qin <37519855+henryqin1997@users.noreply.github.com> Date: Tue, 18 Oct 2022 19:32:19 -0700 Subject: [PATCH] [NFC] polish tests/test_layers/test_3d/test_3d.py code style (#1740) --- tests/test_layers/test_3d/test_3d.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_layers/test_3d/test_3d.py b/tests/test_layers/test_3d/test_3d.py index 9d74c8a57..c79dde2a1 100644 --- a/tests/test_layers/test_3d/test_3d.py +++ b/tests/test_layers/test_3d/test_3d.py @@ -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')