pytorch/torch/cuda
2022-06-01 05:17:41 +00:00
..
amp Resolve TODO after Python 2 for custom_fwd (#78592) 2022-06-01 05:17:41 +00:00
__init__.py Restore old names for private funcs in legacy storages (#77861) 2022-05-20 02:03:34 +00:00
_utils.py
comm.py
error.py
graphs.py Add an argument to specify warmup iterations (#78124) 2022-05-25 01:21:15 +00:00
jiterator.py Fix jiterator doc format (#78471) 2022-05-31 03:44:52 +00:00
memory.py Add __all__ for torch.cuda.memory 2022-04-28 15:06:12 +00:00
nccl.py Avoid collections deprecation warning (#72239) 2022-02-23 02:31:42 +00:00
nvtx.py [PyTorch] Support NVTX range_start and range_end (#70030) 2022-02-07 17:31:57 +00:00
profiler.py
random.py
sparse.py
streams.py reduce overhead of get_current_stream (#78066) 2022-05-23 03:14:01 +00:00