pytorch/torch/testing/_internal/distributed
Pritam Damania 8da7576303 Remove balance and devices parameter from Pipe. (#46804)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46804

As per our design in https://github.com/pytorch/pytorch/issues/44827,
changign the API such that the user places modules on appropriate devices
instead of having a `balance` and `devices` parameter that decides this.

This design allows us to use RemoteModule in the future.
ghstack-source-id: 116479842

Test Plan: waitforbuildbot

Reviewed By: mrshenli

Differential Revision: D24524219

fbshipit-source-id: 9973172c2bb7636572cdc37ce06bf8368638a463
2020-11-12 14:20:23 -08:00
..
nn [RPC Framework] Support remote device format "<workername>/<device>" (#46773) 2020-10-29 00:14:56 -07:00
pipeline Remove balance and devices parameter from Pipe. (#46804) 2020-11-12 14:20:23 -08:00
rpc Support RRef.backward() for Owner RRefs. (#46641) 2020-11-07 21:25:32 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
ddp_under_dist_autograd_test.py [RPC Framework] Support remote device format "<workername>/<device>" (#46773) 2020-10-29 00:14:56 -07:00
distributed_test.py [NCCL] enable p2p tests (#47797) 2020-11-12 10:44:50 -08:00
rpc_utils.py Add Batch-Updating Parameter Server Example to CI Tests (#46510) 2020-10-21 08:46:46 -07:00