Added index 0 for ROCR_VISIBLE_DEVICES (#2268)

BLOrange-AMD · web-flow · commit 13a8686b842b · 2025-06-12T00:08:41.000-05:00
This PR is to fix https://ontrack-internal.amd.com/browse/SWDEV-534855: test_cuda.py::TestCuda::test_hip_device_count fails on one gpu machine. Needs to cherry-pick it to upstream as well once this PR is merged.
diff --git 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},
         ]
 

Original file line number	Diff line number	Diff line change
`@@ -3601,7 +3601,7 @@ def test_hip_device_count(self):`
`3601`	`3601`	`{"CUDA_VISIBLE_DEVICES": "0", "HIP_VISIBLE_DEVICES": None},`
`3602`	`3602`	`{"CUDA_VISIBLE_DEVICES": None, "HIP_VISIBLE_DEVICES": "0"},`
`3603`	`3603`	`{"CUDA_VISIBLE_DEVICES": "0,1,2,3", "HIP_VISIBLE_DEVICES": "0"},`
`3604`		`- {"ROCR_VISIBLE_DEVICES": "1,2,3", "HIP_VISIBLE_DEVICES": "0"},`
	`3604`	`+ {"ROCR_VISIBLE_DEVICES": "0,1,2,3", "HIP_VISIBLE_DEVICES": "0"},`
`3605`	`3605`	`{"ROCR_VISIBLE_DEVICES": "0", "HIP_VISIBLE_DEVICES": None},`
`3606`	`3606`	`]`
`3607`	`3607`