Skip to content

Commit 08fb558

Browse files
authored
[Bugfix][ROCm] Fix import error on ROCm (#17495)
Signed-off-by: Gregory Shtrasberg <[email protected]>
1 parent dbc18e7 commit 08fb558

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/rotary_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
from vllm.model_executor.custom_op import CustomOp
3333
from vllm.platforms import current_platform
3434

35-
if current_platform.is_cuda_alike():
35+
if current_platform.is_cuda():
3636
from vllm.vllm_flash_attn.layers.rotary import apply_rotary_emb
3737

3838

0 commit comments

Comments
 (0)