minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

[Kernel] Classification cleanup: unify _jit_ naming, drop empty/model groups, add elementwise (RFC #29630) (#32148)

Browse Source 
Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Xiaoyu Zhang
2026-07-23 13:47:02 +08:00
committed by GitHub
co-authored by Claude Opus 4.8
parent 1b63155efe
commit 11b0e5c5ad
102 changed files with 173 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/ops/diffusion/test_qknorm_rope.py
+1 -1
View File
@@ -48,7 +48,7 @@ def split_qknorm_rope(
) -> None:
from flashinfer.rope import apply_rope_with_cos_sin_cache_inplace
from sglang.kernels.ops.layernorm._jit_norm import fused_inplace_qknorm
from sglang.kernels.ops.layernorm.norm import fused_inplace_qknorm
fused_inplace_qknorm(q, k, q_weight, k_weight)
apply_rope_with_cos_sin_cache_inplace(