add gloo support for gather on GPU (#14916)
authorJane Wang <janewang@fb.com>
Wed, 12 Dec 2018 05:03:13 +0000 (21:03 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 12 Dec 2018 05:21:10 +0000 (21:21 -0800)
commitf8455ed754196cd554cbaaaaa440dd178383154d
treea5467795d4578ab5b4acb2d32ca2db8b17b8d4ff
parent3fa53da61a82a19193619f48c3bfcf91d5eb20c0
add gloo support for gather on GPU (#14916)

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

as titled

Reviewed By: pietern

Differential Revision: D13267832

fbshipit-source-id: 3b89d08af93f74941f17ff892c33fc2a4a023c19
test/test_c10d.py
torch/lib/c10d/ProcessGroupGloo.cpp