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
__init__.py	[ROCm] Add AMDSMI support for UUID input (#129741)	2024-10-15 15:56:30 +00:00
_gpu_trace.py
_memory_viz.py
_sanitizer.py
_utils.py
comm.py
error.py
gds.py
graphs.py
jiterator.py
memory.py	raw_alloc ignores PYTORCH_NO_CUDA_MEMORY_CACHING (#131114)	2024-10-04 15:36:29 +00:00
nccl.py
nvtx.py
profiler.py
random.py
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