pytorch/torch/cuda
2025-10-01 19:05:53 +00:00
..
amp [AMP] Add deprecated decorator for torch.xxx.amp.autocast class (#163654) 2025-09-27 14:37:12 +00:00
__init__.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_device_limits.py [torch][cuda][device_limits] Library for querying device hardware limits for flops and bandwidth (#162942) 2025-09-23 04:48:19 +00:00
_gpu_trace.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_memory_viz.py [BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554) 2025-07-03 18:56:07 +00:00
_pin_memory_utils.py [dcp] add new checkpoint staging to preserve storage sharing and support mutable state_dicts (#155192) 2025-06-19 02:04:21 +00:00
_sanitizer.py
_utils.py [1/N] Simplify "in" operation for containers of a single item (#164224) 2025-09-30 19:59:43 +00:00
comm.py
gds.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
graphs.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
jiterator.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
memory.py Replace Literal[None] with None in typing (#163489) 2025-09-22 22:10:08 +00:00
nccl.py
nvtx.py
profiler.py
random.py Avoid unnecessary clone in torch.cuda.set_rng_state (#149283) 2025-03-18 20:47:57 +00:00
sparse.py
streams.py Implement CUDA stream protocol (#163614) 2025-09-23 21:02:08 +00:00
tunable.py [ROCm] Remove HIPBLASLT_ALLOW_TF32 from codebase (#162998) 2025-09-18 13:53:48 +00:00