pytorch/torch/cuda
Tal Ben-Nun c0d710634f Respect ROCR_VISIBLE_DEVICES on AMD GPU device discovery (#142292)
Reland of #140320 after failing test on trunk. Fixes potential environment clobbering in test, makes ROCr+HIP devices (if specified together) more robust to index errors.

Fixes #140318

Pull Request resolved: https://github.com/pytorch/pytorch/pull/142292
Approved by: https://github.com/jataylo, https://github.com/huydhn, https://github.com/jeffdaily

Co-authored-by: Jack Taylor <108682042+jataylo@users.noreply.github.com>
Co-authored-by: Jeff Daily <jeff.daily@amd.com>
2024-12-25 02:37:11 +00:00
..
amp
__init__.py Respect ROCR_VISIBLE_DEVICES on AMD GPU device discovery (#142292) 2024-12-25 02:37:11 +00:00
_gpu_trace.py [BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447) 2024-12-23 14:04:00 +00:00
_memory_viz.py [BE][Easy] enable PYFMT for torch/[a-s]*/ (#138447) 2024-12-23 14:04:00 +00:00
_sanitizer.py
_utils.py
comm.py
error.py
gds.py
graphs.py
jiterator.py
memory.py
nccl.py
nvtx.py
profiler.py
random.py
sparse.py
streams.py
tunable.py [ROCm] Fix unit test: matmul_offline_mgpu_tunableop (#143507) 2024-12-19 19:48:20 +00:00