update HIP_VISIBLE_DEVICES of rocm

d0cc5577 · zhuwenwen · 65e29a89 · d0cc5577
Commit d0cc5577 authored Aug 01, 2025 by zhuwenwen
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 6 deletions

vllm/platforms/rocm.py vllm/platforms/rocm.py +6 -6

No files found.
--- a/vllm/platforms/rocm.py
+++ b/vllm/platforms/rocm.py
@@ -75,12 +75,12 @@ _ROCM_DEVICE_ID_NAME_MAP: dict[str, str] = {
 }
 # Prevent use of clashing `{CUDA/HIP}_VISIBLE_DEVICES``
-# if "HIP_VISIBLE_DEVICES" in os.environ:
+if "HIP_VISIBLE_DEVICES" in os.environ:
-#     val = os.environ["HIP_VISIBLE_DEVICES"]
+    val = os.environ["HIP_VISIBLE_DEVICES"]
-#     if cuda_val := os.environ.get("CUDA_VISIBLE_DEVICES", None):
+    if cuda_val := os.environ.get("CUDA_VISIBLE_DEVICES", None):
-#         assert val == cuda_val
+        assert val == cuda_val
-#     else:
+    else:
-#         os.environ["CUDA_VISIBLE_DEVICES"] = val
+        os.environ["CUDA_VISIBLE_DEVICES"] = val
 # AMDSMI utils
 # Note that NVML is not affected by `{CUDA/HIP}_VISIBLE_DEVICES`,