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

2025-12-04 16:45:44 +08:00
parent e3ab23c1a6
commit cb8df87fc1
8 changed files with 293 additions and 12 deletions
@@ -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 (