Nikita Shulga 8ab2ad306d Enable torch.cuda.nccl typechecking (#45344) ... Summary: Fixes https://github.com/pytorch/pytorch/issues/45336 Pull Request resolved: https://github.com/pytorch/pytorch/pull/45344 Reviewed By: walterddr Differential Revision: D23935306 Pulled By: malfet fbshipit-source-id: dd09d4f8ff7a327131764487158675027a13bf69		2020-09-25 17:02:47 -07:00
..
amp	fix typo in GradScaler docstring (#42236)	2020-07-29 13:14:57 -07:00
__init__.py	[minor] Fix undefined variable (#45246)	2020-09-24 20:17:13 -07:00
_utils.py	Decouple DataParallel/DistributedDataParallel from CUDA (#38454)	2020-07-07 12:48:16 -07:00
comm.py	Add torch.cuda.comm to typechecking CI (#45350)	2020-09-25 12:13:43 -07:00
error.py	remediation of S205607	2020-07-17 17:19:47 -07:00
memory.py	adds list_gpu_processes function (#44616)	2020-09-14 09:54:32 -07:00
nccl.py	Enable torch.cuda.nccl typechecking (#45344)	2020-09-25 17:02:47 -07:00
nvtx.py	Enable torch.cuda.nvtx typechecking (#43443)	2020-08-24 08:20:12 -07:00
profiler.py	Move torch.cuda annotations inline (#40075)	2020-06-18 15:52:29 -07:00
random.py	doc string fix for torch.cuda.set_rng_state_all (#40544)	2020-06-25 08:37:14 -07:00
sparse.py	Delete dead Tensor code paths (#5417)	2018-02-27 17:58:09 -05:00
streams.py	Update docs about CUDA stream priority (#41364)	2020-08-05 20:03:18 -07:00