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

[Kernel] Move CUDA and ROCm speculative kernels to JIT (#40033)

Browse Source
This commit is contained in:
Liangsheng Yin
2026-09-17 17:27:26 -07:00
committed by GitHub
parent 20518d8518
commit f65c70bb7d
29 changed files with 1122 additions and 546 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-musa.yml
+3 -3
View File
@@ -222,9 +222,9 @@ jobs:
- name: Run sgl-kernel test
timeout-minutes: 20
run: |
pytest python/sglang/kernels/aot/tests/speculative/test_eagle_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_ngram_utils.py
pytest python/sglang/kernels/aot/tests/speculative/test_speculative_sampling.py
pytest test/registered/kernels/ops/speculative/test_eagle_utils.py
pytest test/registered/kernels/ops/speculative/test_ngram_utils.py
pytest test/registered/kernels/ops/speculative/test_speculative_sampling.py
pytest python/sglang/kernels/aot/tests/test_torch_defaults_reset.py