From fe20ef8052d8f0a06f6d4696dad0d289fb2df8aa Mon Sep 17 00:00:00 2001 From: Bo Li <110066325+BLOrange-AMD@users.noreply.github.com> Date: Wed, 11 Jun 2025 23:06:35 -0500 Subject: [PATCH] Added index 0 for ROCR_VISIBLE_DEVICES --- test/test_cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_cuda.py b/test/test_cuda.py index 83da1f82d42fd..9dd18eb12cfbb 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -3601,7 +3601,7 @@ def test_hip_device_count(self): {"CUDA_VISIBLE_DEVICES": "0", "HIP_VISIBLE_DEVICES": None}, {"CUDA_VISIBLE_DEVICES": None, "HIP_VISIBLE_DEVICES": "0"}, {"CUDA_VISIBLE_DEVICES": "0,1,2,3", "HIP_VISIBLE_DEVICES": "0"}, - {"ROCR_VISIBLE_DEVICES": "1,2,3", "HIP_VISIBLE_DEVICES": "0"}, + {"ROCR_VISIBLE_DEVICES": "0,1,2,3", "HIP_VISIBLE_DEVICES": "0"}, {"ROCR_VISIBLE_DEVICES": "0", "HIP_VISIBLE_DEVICES": None}, ]