Conversation
The QSA decode resolver gated trtllm-gen paged decode on is_sm100_supported(),
so GB10 (sm_121) fell through to the FA4 cute varlen fallback. That fallback
runs a prefill-shaped kernel at decode row counts and aborts in its epilogue:
MLIRError: expects `coord` and shape of view are weakly congruent, but got
'!cute.layout<"(?,?):(?{i64 div=8},1)">', '!cute.coord<"(_,_,?)">'
sm_121 runs the same trtllm-gen decode kernel, so admit it via the existing
is_sm121() helper. Verified on 2x DGX Spark serving Qwen3.8-Flash-Next NVFP4
at tp=2: server reaches ready and decodes correctly; prefill ~1318 tok/s.
cshintov
requested review from
Fridge003,
HaiShaw,
Qiaolin-Yu,
hebiao064,
ispobock and
merrymercy
as code owners
August 27, 2026 08:02
yhyang201
merged commit Aug 27, 2026
7c66045
into
sgl-project:qwen4-main-squashed
80 of 90 checks passed
3 tasks
5 tasks
5 tasks
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
_resolve_trtllm_sparse_decode()gates the trtllm-gen paged decode kernel onis_sm100_supported(). GB10 (DGX Spark, sm_121) fails that check, so QSA decodefalls through to the FA4 cute varlen fallback.
As the resolver's own docstring notes, that fallback "runs a prefill-shaped kernel
at decode row counts". On sm_121 it does not merely run slower, it aborts during
CUDA graph capture:
This makes Qwen3.8-Flash-Next unservable on DGX Spark, because
_forward_paged_attentionis the only decode path QSA has.flashinfer.decode.trtllm_batch_decode_with_kv_cacheimports fine on sm_121, andthe repo already ships an
is_sm121()helper (added for GB10). Admitting sm_121to the gate selects the working kernel.
Modifications
python/sglang/srt/layers/attention/qwen_sparse_attn_backend.py: widen the gatefrom
is_sm100_supported()tois_sm100_supported() or is_sm121(). One line;no behaviour change on any other architecture.
Accuracy Test
Verified on 2x DGX Spark (GB10, sm_121, 128 GB unified each, ConnectX 200 Gb/s)
serving
RadixArk/Qwen3.8-Flash-Next-NVFP4at--tp-size 2 --nnodes 2:Benchmarking and Profiling
Same host,
--mem-fraction-static 0.70,--ple-offload-embedding,--moe-runner-backend flashinfer_cutlass, context 65536:No before-numbers are possible: the server does not start without this change.
Checklist
CI States
Latest PR Test (Base): ❌ Run #33052279089
Latest PR Test (Extra): ❌ Run #33052278925
Latest PR Test (AMD ROCm 7.2): ❌ Run #33052279077