From 5241ca50d671e682c9e7fa2b70b761b0a7a4bc98 Mon Sep 17 00:00:00 2001 From: Gregory Shtrasberg <156009573+gshtras@users.noreply.github.com> Date: Tue, 10 Jun 2025 18:06:15 -0400 Subject: [PATCH] [ROCm][V1] Adding ROCm to the list of plaforms using V1 by default (#19440) 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 4ce1b41e4f87f..f28f7cba46257 100644 --- a/vllm/engine/arg_utils.py +++ b/vllm/engine/arg_utils.py @@ -1442,7 +1442,7 @@ class EngineArgs: # Non-[CUDA, TPU] may be supported on V1, but off by default for now. v0_hardware = not any( - (current_platform.is_cuda(), current_platform.is_tpu(), + (current_platform.is_cuda_alike(), current_platform.is_tpu(), (current_platform.is_cpu() and current_platform.get_cpu_architecture() == CpuArchEnum.X86))) if v0_hardware and _warn_or_fallback( # noqa: SIM103