pytorch/torch/cuda
2020-11-13 08:14:10 -08:00
..
amp Use MTA for amp grad unscaling, enforce op math type in MTA functors, and allow op lambdas (#44778) 2020-10-01 07:51:16 -07:00
__init__.py Merge branch 'cuda_stream_jit' of https://github.com/pytorch/pytorch into cuda_stream_jit 2020-11-13 08:14:10 -08: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 Add nvtx.range() context manager (#42925) 2020-10-22 19:46:16 -07:00
profiler.py Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -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
streams.py Update docs about CUDA stream priority (#41364) 2020-08-05 20:03:18 -07:00