Researched: W4A4 hangs on GB10 because FlashInfer ships no FP4 cubins for sm_120/121 (all datacenter Sm100a); dense mm_fp4 is gated-off/returns-zeros on consumer Blackwell, and the FlashInfer FP4 autotuner spins on the first forward pass. Not a misconfig - dense W4A4 inference isn't validated on sm_121. W4A16 (4-bit weight / 16-bit act, Marlin) vs llama Q4_K_M is the correct apples-to- apples (same quant class) AND the fast path. Removed the misleading 'W4A4 would be faster / lower bound' framing. Sources: vllm #30163/#26381, flashinfer #2577/#3294, cutlass #3096. Assisted-by: Claude:opus-4.8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
7.3 KiB
Paged-attention / parity benchmarks (GB10 / DGX Spark)
Goal of the series: vLLM parity. This records the measured gap so the parity claim is data-backed, not asserted.
Setup: GB10 (sm_121, 119 GiB unified). Model Qwen3-Coder-30B-A3B. llama.cpp = pinned base + this series
(MXFP4_MOE, -fa 1 -b 2048 -ub 2048, llama-batched-bench, PP=512 TG=128). vLLM = 0.23.0 FP8 (recorded
prior run, same box/model). S_PP / S_TG are aggregate prefill / decode tok/s across B streams.
Fresh llama.cpp (this series, MXFP4) vs vLLM (FP8)
| B | llama S_PP | vLLM S_PP | PP gap | llama S_TG | vLLM S_TG | TG gap |
|---|---|---|---|---|---|---|
| 1 | 1565 | 9644 | 6.2× | 83 | 48 | llama wins |
| 8 | 3648 | 33373 | 9.1× | 126 | 312 | 2.5× |
| 32 | 2074 | 99398 | 48× | 319 | 1171 | 3.7× |
| 64 | 3643 | 151990 | 42× | 771 | 2064 | 2.7× |
Verdict — two distinct gaps, only one is the engine's
- Prefill (S_PP): 6–48× behind, and it does NOT scale with B (plateaus ~3.6k). This is the FP4 MoE
GEMM kernel (
mul_mat_q<MXFP4>~22 TFLOP/s), confirmed earlier. Paged attention cannot close this — it's per-token compute. Needs the tcgen05/CUTLASS grouped-GEMM (Lever 3, multi-week, no upstream base). - Decode at concurrency (S_TG): 2.5–3.7× behind for B≥8 (we win at B=1). This gap IS partly the engine's domain — vLLM's block-paged KV + continuous batching pack more concurrent decode work per step. This is what patches 0003–0006 target. The win here is realistic; the prefill win is not (kernel).
CORRECTION — decode-phase profile (B=64, decode-dominated nsys)
The "decode gap is engine-addressable" read above was wrong. Profiling a decode-dominated B=64 run:
| kernel | % GPU time |
|---|---|
mul_mat_q<MXFP4> (MoE GEMM) |
54.6 |
flash_attn_ext (attention) |
19.8 |
mul_mat_q<Q8> (dense) |
10.9 |
| KV writes / quant / norms / rest | ~15 |
Decode at concurrency is ALSO dominated by the FP4 MoE GEMM (54.6%) — the same Lever-3 kernel as prefill. Attention (the only thing paging optimizes) is ~20%, and the gather-read reclaims only the masked-cell fraction of that. So the paged series (0003–0006) cannot close the vLLM gap in either phase — both are MoE-kernel-bound. vLLM's concurrency advantage is its MoE/attention kernels, not (mainly) its KV management.
What the paged series IS still good for (just not throughput parity)
- Capacity: block-granular + on-demand allocation → fit more/longer concurrent sequences in fixed VRAM.
- Prefix sharing: cross-request block dedup → lower TTFT + memory on shared system prompts / RAG.
These are real wins on memory-pressured and shared-prefix workloads — but they are not tok/s parity, and batched-bench (fresh, non-fragmented, no shared prefix) won't show them.
DENSE model parity (Qwen3-32B) — does the kernel gap exist for dense too? YES.
The MoE work above is about the grouped MoE GEMM. Dense models use a different (non-grouped) matmul path, so we benchmarked a dense 32B head-to-head.
Headline comparison — vLLM NVFP4 W4A16 vs llama.cpp Q4_K_M. This is the correct apples-to-apples on
DGX Spark: both are 4-bit weights / 16-bit activations (same quant class). vLLM = Qwen3-32B-NVFP4A16
(FlashInfer Marlin W4A16 kernel); llama.cpp = Qwen3-32B-Q4_K_M (int8-MMQ compute). The only difference is
the compute kernel — which is exactly what we're measuring. (Full W4A4 NVFP4 does not run on GB10 today;
root cause below — and it would not be a fair comparison even if it did, since Q4_K_M is also weight-only-4-bit.)
| B | llama Q4_K_M PP | vLLM W4A16 PP | PP gap | llama decode | vLLM decode | TG gap |
|---|---|---|---|---|---|---|
| 1 | 708 | 5367 | 7.6× | 10.2 | 11.7 | ~parity |
| 8 | 761 | 14941 | 20× | 58 | 92 | 1.6× |
| 32 | 763 | 21952 | 29× | 205 | 330 | 1.6× |
| 64 | 765 | 24444 | 32× | 253 | 569 | 2.2× |
Findings:
- Dense prefill has the SAME (larger) kernel gap. llama dense prefill plateaus at ~765 t/s regardless of B; vLLM scales to 24.4k (32×). Both read 4-bit weights — the gap is the compute kernel: vLLM's FP4 Marlin tensor-core GEMM vs llama's int8-MMQ. (Note: on consumer Blackwell, W4A16 Marlin is also reported faster than the experimental W4A4 path, so W4A16 isn't a handicapped stand-in — it's the fast path.)
- Decode is ~parity at B=1 (10.2 vs 11.7 — both weight-bandwidth-bound reading 4-bit weights), and the gap grows with batch (compute starts to matter → the kernel gap reappears: 2.2× at B=64).
- Scope decision (the reason for this benchmark): the Lever-3 kernel track must also deliver a NON-grouped
block-scaled FP4 GEMM for dense, not only the MoE grouped GEMM. The dense GEMM is the simpler of the two
(a plain CUTLASS dense GEMM), so it's a good first kernel to land — and it benefits every dense model.
- No cheap lever:
GGML_CUDA_FORCE_CUBLASis a no-op for dense too (Q4_K pp512: 720.8 vs 721.8) — dequant→cuBLAS-BF16 doesn't engage / isn't faster than int8-MMQ on GB10. With ubatch (saturates) and nwarps (static_assert) already ruled out for MoE, every config/flag lever is now exhausted for both model classes. Parity is strictly the FP4 tensor-core kernel.
- No cheap lever:
- Why full W4A4 NVFP4 hangs on GB10 (root cause, researched). This is a known consumer-Blackwell
limitation, not a misconfiguration. FlashInfer ships no FP4 cubins for sm_120/sm_121 — its precompiled
kernels are all datacenter
Sm100a/Sm103a(B200/B300). So on GB10 the densemm_fp4W4A4 GEMM has no working kernel: the optimized path is gated off for sm_121 (heuristic checksminor==0; 12.1 fails), the CUTLASS dense FP4 fallback is documented to silently return all-zeros, and TRT-LLM errors at capability 120. Our exact symptom — loads weights, then stalls at the first profiling forward pass withenable_flashinfer_autotune=Trueat 0–3% GPU — is the FlashInfer FP4 autotuner/JIT spinning on an arch with no FP4 cubins (matches vllm #30163/#26381, flashinfer #2577/#3294). The "NVFP4 on DGX Spark" story everyone cites is about quantization + memory footprint + W4A16/MoE, not dense W4A4 inference, which isn't validated on sm_121 yet (where people patched it working, it was slower than W4A16 anyway). Therefore W4A16 vs Q4_K_M above is the right, reproducible apples-to-apples for DGX Spark today. Optional W4A4 retry (verify output isn't zeros first):VLLM_SKIP_FLASHINFER_AUTOTUNE=1+VLLM_NVFP4_GEMM_BACKEND=cutlass+--enforce-eager, or NVIDIA'svllm/vllm-openai:cu130-nightlycontainer.
So, honestly, where parity stands
- Decode single-stream: already at/above parity (B=1: 83 vs 48).
- Decode concurrency: a real, engine-addressable gap the paged series can narrow (0004 on-demand pool + 0005 continuous batching). Target: close the 2.5–3.7× at B≥8.
- Prefill: kernel-bound, not engine-bound. No amount of paging reaches vLLM here; that's a separate track.
Series status when measured: 0001 (vendor) + 0002 (placement, token-identical) done; 0003 (gather-read) turn-key-planned, not yet implemented. These numbers are the baseline the engine patches must improve on at B≥8 decode — re-run this table after 0004/0005 to show the concurrency gap closing.