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

[Kernel] Move sgl-kernel under sglang.kernels.aot (#32648)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2026-07-29 17:25:00 +08:00
committed by GitHub
parent 1b9dfa14e6
commit c32c4ef79c
370 changed files with 300 additions and 269 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/cpu/test_spec_kernels.py
+2 -2
View File
@@ -294,7 +294,7 @@ class TestVerifyTreeGreedy(CustomTestCase):
def test_verify_tree_greedy_upstream_golden(self):
# Golden fixture ported from the CUDA kernel UT
# sgl-kernel/tests/speculative/test_eagle_utils.py::test_verify_tree_greedy
# python/sglang/kernels/aot/tests/speculative/test_eagle_utils.py::test_verify_tree_greedy
# (device swapped to CPU); expected outputs are the CUDA kernel's.
candidates = torch.tensor(
[
@@ -767,7 +767,7 @@ class TestReconstructIndicesFromTreeMask(CustomTestCase):
bs, draft_token_num = 2, 4
seq_lens = torch.tensor([12, 5], dtype=torch.int64)
# Request 0: root(0) -> {1, 2}, 2 -> 3 (golden case from
# sgl-kernel/tests/speculative/test_ngram_utils.py).
# python/sglang/kernels/aot/tests/speculative/test_ngram_utils.py).
# Request 1: plain chain 0 -> 1 -> 2 -> 3.
tree_mask = torch.tensor(
# fmt: off