From c7e63aa4d84de4f0b076d2974d30cd1cd34a4191 Mon Sep 17 00:00:00 2001 From: Gregory Shtrasberg <156009573+gshtras@users.noreply.github.com> Date: Tue, 1 Apr 2025 03:10:48 -0400 Subject: [PATCH] [ROCm] Use device name in the warning (#15838) Signed-off-by: Gregory Shtrasberg --- vllm/engine/arg_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/engine/arg_utils.py b/vllm/engine/arg_utils.py index e29b04ab6e0c8..019cbe18397e6 100644 --- a/vllm/engine/arg_utils.py +++ b/vllm/engine/arg_utils.py @@ -1530,7 +1530,7 @@ class EngineArgs: # Non-CUDA is supported on V1, but off by default for now. not_cuda = not current_platform.is_cuda() if not_cuda and _warn_or_fallback( # noqa: SIM103 - current_platform.device_type): + current_platform.device_name): return False #############################################################