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

[Kernel] Migrate DSA + DSV4 attention kernels to sglang.kernels (RFC #29630, Phase 2.5, 5/7) (#30792)

Browse Source 
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
Xiaoyu Zhang
2026-07-15 11:11:22 +08:00
committed by GitHub
co-authored by Claude Fable 5
parent 4ae9cc3c81
commit ba5be86d42
60 changed files with 663 additions and 583 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/manual/layers/attention/dsa/test_act_quant_triton.py
+2 -2
View File
@@ -9,8 +9,8 @@ from typing import Tuple
import pytest
import torch
from sglang.srt.layers.attention.dsa.tilelang_kernel import act_quant
from sglang.srt.layers.attention.dsa.triton_kernel import act_quant as act_quant_triton
from sglang.kernels.ops.attention.dsa.tilelang_kernel import act_quant
from sglang.kernels.ops.attention.dsa.triton_kernel import act_quant as act_quant_triton
def benchmark_kernel(
test/manual/layers/attention/dsa/test_get_k_scale_triton_kernel.py
+1 -1
View File
@@ -1,6 +1,6 @@
import torch
from sglang.srt.layers.attention.dsa.index_buf_accessor import (
from sglang.kernels.ops.attention.dsa.index_buf_accessor import (
_get_k_and_s_triton_kernel,
)
test/manual/layers/attention/dsa/test_index_buf_accessor.py
+1 -1
View File
@@ -13,7 +13,7 @@ Test coverage:
import pytest
import torch
from sglang.srt.layers.attention.dsa.index_buf_accessor import GetK, GetKAndS, GetS
from sglang.kernels.ops.attention.dsa.index_buf_accessor import GetK, GetKAndS, GetS
class MockDSATokenToKVPool: