minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 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/manual/layers/test_fused_gate_sigmoid_mul_add.py
+1 -1
View File
@@ -3,7 +3,7 @@ import itertools
import pytest
import torch
from sglang.kernels.ops.layernorm.elementwise import fused_gate_sigmoid_mul_add
from sglang.kernels.ops.elementwise.elementwise import fused_gate_sigmoid_mul_add
DTYPES = [torch.float16, torch.bfloat16]
TOKEN_COUNTS = [1, 2, 4, 8, 16, 64, 512, 1024, 2048, 4096, 8192]
test/manual/layers/test_fused_sigmoid_mul.py
+1 -1
View File
@@ -3,7 +3,7 @@ import itertools
import pytest
import torch
from sglang.kernels.ops.layernorm.elementwise import fused_sigmoid_mul
from sglang.kernels.ops.elementwise.elementwise import fused_sigmoid_mul
DTYPES = [torch.float16, torch.bfloat16]
TOKEN_COUNTS = [1, 2, 4, 8, 16, 64, 512, 1024, 2048, 4096, 8192]