Tongzhou Wang 8e33451e2e Make torch.cuda.* take device objects; Update distributed docs (#10833) ... Summary: Commits: 1. Make `torch.cuda.*` take device objects 2. Update `torch.distributed` docs to emphasize calling `torch.cuda.set_device` before `init_process_group` Pull Request resolved: https://github.com/pytorch/pytorch/pull/10833 Differential Revision: D9514241 Pulled By: SsnL fbshipit-source-id: 2497464305fb1e63d6c495291a5744aaa7e2696e		2018-08-27 15:24:42 -07:00
..
__init__.py	Make torch.cuda.* take device objects; Update distributed docs (#10833)	2018-08-27 15:24:42 -07:00
_utils.py	Make torch.cuda.* take device objects; Update distributed docs (#10833)	2018-08-27 15:24:42 -07:00
comm.py	Move nccl scatter and gather to C++ (#9117)	2018-07-06 11:10:33 -07:00
error.py	Add CUDA profiler bindings	2017-09-25 23:21:30 -04:00
nccl.py	Add streams and comms as optional arguments (#3968)	2017-12-04 13:51:22 -05:00
nvtx.py	Improve Windows Compatibility (for csrc/scripts) (#2941)	2017-11-08 19:51:35 +01:00
profiler.py	Fix nvprof mode in autograd profiler	2017-10-20 10:22:54 -04:00
random.py	Merge Variable and Tensor classes (#5225)	2018-02-23 18:03:31 -05:00
sparse.py	Delete dead Tensor code paths (#5417)	2018-02-27 17:58:09 -05:00
streams.py	Make torch.cuda.* take device objects; Update distributed docs (#10833)	2018-08-27 15:24:42 -07:00