OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ecc5e05854
pytorch/torch/cuda
History
Jack Taylor 966a1a971e [ROCm] Add AMDSMI support for UUID input (#129741) 
Adds support for for using UUIDs for AMDSMI utilities in PyTorch via CUDA_VISIBLE_DEVICES/HIP_VISIBLE_DEVICES.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/129741
Approved by: https://github.com/pruthvistony, https://github.com/jeffdaily
2024-10-15 15:56:30 +00:00
..
amp [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
__init__.py [ROCm] Add AMDSMI support for UUID input (#129741) 2024-10-15 15:56:30 +00:00
_gpu_trace.py Refactor gpu trace to be device-agnostic (#121794) 2024-03-30 13:04:38 +00:00
_memory_viz.py Flip default value for mypy disallow_untyped_defs [4/11] (#127841) 2024-06-08 18:36:48 +00:00
_sanitizer.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
_utils.py refactor code to share across different devices (#120602) 2024-02-28 09:42:58 +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
error.py
gds.py [Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489) 2024-08-15 17:11:52 +00:00
graphs.py fixes #124582 (#128483) 2024-07-02 08:45:59 +00:00
jiterator.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
memory.py raw_alloc ignores PYTORCH_NO_CUDA_MEMORY_CACHING (#131114) 2024-10-04 15:36:29 +00:00
nccl.py Flip default value for mypy disallow_untyped_defs [4/11] (#127841) 2024-06-08 18:36:48 +00:00
nvtx.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
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 [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
sparse.py
streams.py Use torch.Stream&torch.Event for Dynamo capature (#134850) 2024-10-02 14:15:33 +00:00
tunable.py [ROCm] Tunableop record untuned (#128813) 2024-10-09 21:59:03 +00:00