Yuanyuan Chen 48b54b45d6 Replace pynvml with nvidia-ml-py in win-test.sh (#164681) ...

pynvml was deprecated.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164681
Approved by: https://github.com/Aidyn-A, https://github.com/eqy

2025-10-06 21:57:26 +00:00

..

amp

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

__init__.py

Replace pynvml with nvidia-ml-py in win-test.sh (#164681)

2025-10-06 21:57:26 +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

[2/N] Fix ruff warnings (#164460)

2025-10-04 03:40:32 +00:00

_utils.py

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

comm.py

[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)

2024-08-04 10:24:09 +00:00

gds.py

[4/N] Apply ruff UP035 rule to python code (#164206)

2025-10-01 19:05:53 +00:00

graphs.py

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +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 pynvml with nvidia-ml-py in win-test.sh (#164681)

2025-10-06 21:57:26 +00:00

nccl.py

PEP585 update - torch/_higher_order_ops torch/_subclasses torch/backends torch/compiler torch/cuda torch/masked torch/mtia torch/nested (#145202)

2025-01-20 22:37:26 +00:00

nvtx.py

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

profiler.py

[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)

2024-08-04 10:24:09 +00:00

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