[1/2] Add rope kernel in sgl-kernel (#14334)

2025-12-04 00:45:44 -08:00
parent e3ab23c1a6
commit cb8df87fc1
8 changed files with 294 additions and 13 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -30,6 +30,7 @@ from sgl_kernel.elementwise import (
    gemma_fused_add_rmsnorm,
    gemma_rmsnorm,
    rmsnorm,
+    rotary_embedding,
    silu_and_mul,
 )
 from sgl_kernel.expert_specialization import (