Nikitha Malgi 616503ac19 Merge branch 'cuda_stream_jit' of https://github.com/pytorch/pytorch into cuda_stream_jit		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