projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b9c20d5) | patch
Revert replicate.py to disallow replicating multi-device modules (#19278)
authorShen Li <shenli@fb.com>
Tue, 16 Apr 2019 16:35:36 +0000 (09:35 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 16 Apr 2019 17:03:38 +0000 (10:03 -0700)
commit344acaa0caa91d3acb90b2e1f89be53d153dadc1
tree9be51f5cb3bbe5e4b5f1148c093c4898509f9e9atree | snapshot
parentb9c20d5224caea1f360285f765d351014fd29df5commit | diff
Revert replicate.py to disallow replicating multi-device modules (#19278)

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

Based on discussion in https://github.com/pytorch/pytorch/pull/19278 and https://github.com/pytorch/pytorch/pull/18687, changes to replicate.py will be reverted to disallow replicating multi-device modules.

Reviewed By: pietern

Differential Revision: D14940018

fbshipit-source-id: 7504c0f4325c2639264c52dcbb499e61c9ad2c26
test/common_cuda.py diff | blob | history
test/test_nn.py diff | blob | history
torch/nn/parallel/replicate.py diff | blob | history
Domain: Machine Learning / ML Framework;