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

[Kernel] Migrate top-level srt/layers stray kernels to sglang.kernels (RFC #29630, Phase 2.5, 3/7) (#30787)

Browse Source 
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
Xiaoyu Zhang
2026-07-14 09:20:59 +08:00
committed by GitHub
co-authored by Claude Fable 5
parent e489685509
commit e9ef06c560
37 changed files with 269 additions and 177 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.srt.layers.elementwise import fused_gate_sigmoid_mul_add
from sglang.kernels.ops.layernorm.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.srt.layers.elementwise import fused_sigmoid_mul
from sglang.kernels.ops.layernorm.elementwise import fused_sigmoid_mul
DTYPES = [torch.float16, torch.bfloat16]
TOKEN_COUNTS = [1, 2, 4, 8, 16, 64, 512, 1024, 2048, 4096, 8192]