[ROCm] Begin Deprecation Window for CUDA_VISIBLE_DEVICES on ROCm (#46636)
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
This commit is contained in:
@@ -119,6 +119,14 @@ def _sync_hip_cuda_env_vars():
|
||||
hip_val = os.environ.get("HIP_VISIBLE_DEVICES") or None
|
||||
cuda_val = os.environ.get("CUDA_VISIBLE_DEVICES") or None
|
||||
|
||||
if cuda_val is not None:
|
||||
logger.warning_once(
|
||||
"Using CUDA_VISIBLE_DEVICES on ROCm is deprecated and support "
|
||||
"will be removed in vLLM v0.26.0. Please use HIP_VISIBLE_DEVICES "
|
||||
"instead.",
|
||||
scope="process",
|
||||
)
|
||||
|
||||
if hip_val is not None and cuda_val is not None:
|
||||
if hip_val != cuda_val:
|
||||
raise ValueError(
|
||||
|
||||
Reference in New Issue
Block a user