Commit 10f9349e authored by Myle Ott's avatar Myle Ott Committed by Facebook Github Bot
Browse files

Remove extraneous call to RNG in multi-GPU code path

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/865

Differential Revision: D17510276

Pulled By: myleott

fbshipit-source-id: 24119402ad5fe95a1312fadb77bafe49a9197c6b
parent e869c80d
......@@ -87,7 +87,7 @@ def distributed_init(args):
socket.gethostname(), args.distributed_rank), flush=True)
# perform a dummy all-reduce to initialize the NCCL communicator
dist.all_reduce(torch.rand(1).cuda())
dist.all_reduce(torch.zeros(1).cuda())
suppress_output(is_master(args))
......
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