pytorch/torch/cuda
SsnL dce3d74dfb add torch.cuda.synchronize(device=None) (#19573)
Summary:
fixes https://github.com/pytorch/pytorch/issues/19509
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19573

Differential Revision: D15045730

Pulled By: ezyang

fbshipit-source-id: 732721b4b360fc4348ca7c87d4cd1386e7651bdd
2019-04-23 08:40:38 -07:00
..
__init__.py add torch.cuda.synchronize(device=None) (#19573) 2019-04-23 08:40:38 -07:00
__init__.pyi add torch.cuda.synchronize(device=None) (#19573) 2019-04-23 08:40:38 -07:00
_utils.py Fix trivial typos in torch.cuda._utils (#16026) 2019-01-17 10:40:43 -08:00
comm.py Turn on F401: Unused import warning. (#18598) 2019-03-30 09:01:17 -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 Update cuda.get/set_rng_state doc (#14324) 2018-12-27 14:09:25 -08:00
sparse.py Delete dead Tensor code paths (#5417) 2018-02-27 17:58:09 -05:00
streams.py Unify device() return type in Stream, Event, and Tensor (#16150) 2019-01-19 23:01:31 -08:00