Cleanup init_process_group (#19033)
authorKutta Srinivasan <kutta@fb.com>
Thu, 18 Apr 2019 16:31:03 +0000 (09:31 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 18 Apr 2019 16:37:38 +0000 (09:37 -0700)
commitb7323a94ad02f4de5cdf66cb82284ba43d27e517
tree114b647dd1f624a786444ef5c124676bd8849436
parent20a5aa967049bdb74e35f4e495a9a0f1de1c91a1
Cleanup init_process_group (#19033)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19033

torch.distributed.init_process_group() has had many parameters added, but the contract isn't clear. Adding documentation, asserts, and explicit args should make this clearer to callers and more strictly enforced.

Reviewed By: mrshenli

Differential Revision: D14813070

fbshipit-source-id: 80e4e7123087745bed436eb390887db9d1876042
torch/distributed/distributed_c10d.py