Skip to content

GitLab

Commit b9084e90 authored by Jiezhong Qiu's avatar Jiezhong Qiu
Browse files

Merge branch 'master' of github.com:xptree/poly-xfmr

parents 6b38250f 46e6342e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch/cuda/moe_test.py
View file @ b9084e90
...@@ -159,7 +159,7 @@ def test_dp(): ...@@ -159,7 +159,7 @@ def test_dp():
if __name__ == '__main__': if __name__ == '__main__':
torch.distributed.init_process_group(backend='mpi') torch.distributed.init_process_group(backend='mpi')
world_size = torch.distributed.get_world_size() world_size = torch.distributed.get_world_size()
if len(sys.argv) == 2: if len(sys.argv) >= 2:
task = sys.argv[1] task = sys.argv[1]
print('Specificed task {}'.format(task)) print('Specificed task {}'.format(task))
if task == 'correctness': if task == 'correctness':
......
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