update rocm.py

386e5c47 · zhuwenwen · e3cacb3b · 386e5c47
Commit 386e5c47 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`,