From 08fb5587b4c4c1b031c08adf1b8608d710ab2585 Mon Sep 17 00:00:00 2001 From: Gregory Shtrasberg <156009573+gshtras@users.noreply.github.com> Date: Wed, 30 Apr 2025 22:51:42 -0400 Subject: [PATCH] [Bugfix][ROCm] Fix import error on ROCm (#17495) Signed-off-by: Gregory Shtrasberg --- vllm/model_executor/layers/rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/rotary_embedding.py b/vllm/model_executor/layers/rotary_embedding.py index b179a0f00be2e..8cad5482d6fcd 100644 --- a/vllm/model_executor/layers/rotary_embedding.py +++ b/vllm/model_executor/layers/rotary_embedding.py @@ -32,7 +32,7 @@ from transformers import PretrainedConfig from vllm.model_executor.custom_op import CustomOp from vllm.platforms import current_platform -if current_platform.is_cuda_alike(): +if current_platform.is_cuda(): from vllm.vllm_flash_attn.layers.rotary import apply_rotary_emb