tiny update: use rope kernel in sgl-kernel for amd (#14955)

2025-12-12 18:48:12 -08:00
parent 267170bf1d
commit 77873343c4
1 changed files with 1 additions and 1 deletions
@@ -118,7 +118,7 @@ class RotaryEmbedding(CustomOp):
            and not (_is_cpu and _is_cpu_amx_available)
            and not (_is_xpu)
        ):
-            if _is_cuda:
+            if _is_cuda or _is_hip:
                from sgl_kernel import rotary_embedding
            else:
                from vllm._custom_ops import rotary_embedding