From 77873343c495c0604b4483080ec772b882b14cf0 Mon Sep 17 00:00:00 2001 From: Qiaolin Yu Date: Fri, 12 Dec 2025 18:48:12 -0800 Subject: [PATCH] tiny update: use rope kernel in sgl-kernel for amd (#14955) --- python/sglang/srt/layers/rotary_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/rotary_embedding.py b/python/sglang/srt/layers/rotary_embedding.py index 9675a7e56..e5b129f24 100644 --- a/python/sglang/srt/layers/rotary_embedding.py +++ b/python/sglang/srt/layers/rotary_embedding.py @@ -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