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

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
@@ -7,7 +7,7 @@ over the Qwen3.5 MoE target hidden size.
import torch
import triton
from sglang.kernels.ops.layernorm.elementwise import fused_gate_sigmoid_mul_add
from sglang.kernels.ops.elementwise.elementwise import fused_gate_sigmoid_mul_add
HIDDEN_DIMS = [4096]
+1 -1
View File
@@ -10,7 +10,7 @@ a fair comparison — the reshape/contiguous cost is included.
import torch
import triton
from sglang.kernels.ops.layernorm.elementwise import fused_sigmoid_mul
from sglang.kernels.ops.elementwise.elementwise import fused_sigmoid_mul
NUM_HEADS = 32
HEAD_DIM = 256