[AMD] Enable GLM DSA prefill top-k to the v2 kernel - #37889

Merged

HaiShaw merged 24 commits into

sgl-project:mainfrom

EricKing626:amd/topk-v2-prefill

Sep 21, 2026

Commits on Sep 4, 2026

[AMD] Let the DSA top-k v2 kernel address packed score rows
EricKing626
committed
5d929de
[AMD] Route GLM DSA prefill top-k to the v2 kernel
EricKing626
committed
459ac47
Merge branch 'main' into amd/topk-v2-prefill
1am9trash
authored
ee96b86

Commits on Sep 6, 2026

[AMD] Align the packed-row read base in the paged top-k v2 kernel
EricKing626
committed
b6ae8cf

Commits on Sep 7, 2026

Merge branch 'amd/topk-v2-prefill' of https://github.com/EricKing626/sglang into amd/topk-v2-prefill
EricKing626
committed
c7a5836

Commits on Sep 8, 2026

[AMD] Gate the paged top-k v2 packed-row path on ROCm
EricKing626
committed
960ae11

Commits on Sep 9, 2026

Reject packed rows without a page table
EricKing626
committed
e48e4d3
Check that score rows do not overlap on the packed path
EricKing626
committed
88c15a1
Document that packed rows modify scores in place
EricKing626
committed
b0ccf96

Commits on Sep 11, 2026

Merge branch 'main' into amd/topk-v2-prefill
EricKing626
authored
74034c2

Commits on Sep 14, 2026

Fold the alignment residue into bias instead of a second field
EricKing626
committed
ffcde4e
Merge upstream/main into amd/topk-v2-prefill
EricKing626
committed
c195030
Merge branch 'main' into amd/topk-v2-prefill
EricKing626
authored
8e8808e
Apply clang-format to the raw_indices RuntimeCheck
EricKing626
committed
20a4c00
Trim the comments added by this PR to 1-2 lines
EricKing626
committed
23777de
Leave topk_impl.cuh untouched by this PR
EricKing626
committed
4bbbd19

Commits on Sep 16, 2026

Merge upstream/main into amd/topk-v2-prefill
EricKing626
committed
87d9b23
Merge branch 'main' into amd/topk-v2-prefill
EricKing626
authored
8cf39e6

Commits on Sep 17, 2026

dsv4: split packed-row top-k into its own kernel
EricKing626
committed
fb2eba8
dsv4: keep the packed top-k kernel behind USE_ROCM
EricKing626
committed
e5da6af
dsv4: trim packed-path comments to two lines
EricKing626
committed
0917b10
dsv4: format the packed-test skipif markers
EricKing626
committed
972fd39
Merge branch 'main' into amd/topk-v2-prefill
EricKing626
authored
116560c
route chunked DSA extend through the packed top-k kernel
EricKing626
committed
a6d769e