Wei Wang 7163dce1e0 [CUDA] Compare major version of the runtime device arch against the built version of the pytorch binary (#161299) ... Fixes misleading warning messages when running on sm12x devices using binaries built with sm120. PyTorch binary built with sm120 is compatible with e.g. sm121, so no need for the warning of incompatibility. Also allow the 'matched_cuda_warn' message to show when e.g. the user is running a binary built with only sm90 on sm12x, so that the user would be prompted to get a build which supports e.g. sm120. Pull Request resolved: https://github.com/pytorch/pytorch/pull/161299 Approved by: https://github.com/eqy, https://github.com/atalman		2025-09-24 23:59:19 +00:00
..
amp
__init__.py	[CUDA] Compare major version of the runtime device arch against the built version of the pytorch binary (#161299)	2025-09-24 23:59:19 +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
_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	Use cuda nvrtc so file based on cuda version used by torch (#163642)	2025-09-24 14:23:39 +00:00
comm.py
gds.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +00:00
graphs.py	[CUDAGraph] Add getter for cuda graph exec (#161294)	2025-08-25 20:57:37 +00:00
jiterator.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +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
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