[2/n jit_kernel restruct] unify rotary embedding entrypoints under rope.py (#20247)

2026-03-10 17:49:57 +08:00
parent 6407891b4f
commit 51d9d34977
5 changed files with 63 additions and 90 deletions
--- a/python/sglang/jit_kernel/benchmark/bench_rope.py
+++ b/python/sglang/jit_kernel/benchmark/bench_rope.py
@@ -70,7 +70,7 @@ def sglang_pos_enc_rope(
    positions: torch.Tensor,
    is_neox: bool,
 ) -> None:
-    from sglang.jit_kernel.pos_enc import rotary_embedding_with_key
+    from sglang.jit_kernel.rope import rotary_embedding_with_key

    head_size = q.shape[-1]
    rotary_embedding_with_key(