From 26b5f7bd2a4005dccb797804c93cbce329253003 Mon Sep 17 00:00:00 2001 From: TJian Date: Fri, 1 Aug 2025 05:25:20 -0700 Subject: [PATCH] [BUG] [ROCm] Fix import bug on ROCm (#22083) Signed-off-by: tjtanaa --- vllm/compilation/pass_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/compilation/pass_manager.py b/vllm/compilation/pass_manager.py index 11e03daced160..54f00d5415216 100644 --- a/vllm/compilation/pass_manager.py +++ b/vllm/compilation/pass_manager.py @@ -7,7 +7,7 @@ from vllm.config import VllmConfig from vllm.logger import init_logger from vllm.platforms import current_platform -if current_platform.is_cuda_alike(): +if current_platform.is_cuda(): from .fusion import FusionPass from .collective_fusion import AllReduceFusionPass, AsyncTPPass from .fusion_attn import AttnFusionPass