handle scenario when GPU support is not available and p2p_access_pattern is empty...
authorDeepali Chourasia <deepali@us.ibm.com>
Tue, 19 Mar 2019 06:06:03 +0000 (23:06 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 19 Mar 2019 06:11:54 +0000 (23:11 -0700)
commit542c273e5b1739dc09582dfbf3f801a385105532
tree03b5bfab068848dd0d00676a04a7ebada1fd6413
parent195cba500f2c98c10f3d4b8fc3cca6493099b6df
handle scenario when GPU support is not available and p2p_access_pattern is empty (#17974)

Summary:
Observed that when there is no GPU support available `workspace `sets `GetGpuPeerAccessPattern `to `[]` in
https://github.com/pytorch/pytorch/blob/master/caffe2/python/workspace.py#L79
and this case is not handled in https://github.com/pytorch/pytorch/blob/master/caffe2/python/data_parallel_model.py#L1065.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17974

Differential Revision: D14517066

Pulled By: ezyang

fbshipit-source-id: 186911d95c07e9a55ab82a41d0c7c919e4281bb4
caffe2/python/data_parallel_model.py