[Kernel][SM70] Scale NVFP4 E4M3 decode batches - #282
Conversation
|
CI note: the repository-wide For this head, the changed-file pre-commit suite passes locally in full (Ruff check/format, typos, clang-format, Markdown, mypy, SPDX, config/docs gates), along with 36 focused CPU tests and 10 focused SM70 GPU tests. The PR remains Draft pending the queued post-rebase TP4 full-model smoke. |
Signed-off-by: yangzhuxinyzx <153831768+yangzhuxinyzx@users.noreply.github.com>
Signed-off-by: yangzhuxinyzx <153831768+yangzhuxinyzx@users.noreply.github.com>
Keep the original E4M3 converter on B1 and non-optimized XQA paths, prevent Flash-V100-only E4M3 graph variants under an explicit FlashInfer backend, and record the post-rebase build, regression, and SASS audit evidence. Signed-off-by: yangzhuxinyzx <153831768+yangzhuxinyzx@users.noreply.github.com>
b7ad29d to
5d05b74
Compare
|
Final audit note: |
Purpose
Scale two exact SM70 inference-engine operator contracts without model/checkpoint/architecture identity checks:
Qwen3.8-27B NVFP4 is the matched performance and quality workload only; it does not participate in runtime admission. Gates use SM70 capability, FP8 format, batch/tensor shape, paged-KV layout, graph context, and sampler settings.
Rollback switches:
VLLM_FLASH_V100_E4M3_BATCH_XQA=0restores scalar paged attention for B2-B16.VLLM_FLASH_V100_E4M3_BATCH_XQA_OPTIMIZED=0keeps XQA but restores scalar KV loads, the original E4M3 converter, and baseline CTA routing. B1 always keeps the original converter.VLLM_SM70_TOPK_TOPP_B8_B16_8_WARPS=0restores the prior Triton launch heuristic.Base:
onecat/main@f6a5b57b645867d87f83ada43231f2dd25b40a4aHead:
5d05b74bb56f9d665b0f5c4788df3b17ef1fa6d3Audit fixes
git range-diffequivalent after rebase.FLASHINFER_SM70is explicitly selected.Performance contract
Four V100-SXM2-32GB, TP4, compressed-tensors NVFP4 weights, FP16 activations, FP8-E4M3 KV, Flash-V100, prefix cache, Mamba align, no speculation, FULL decode graphs. Official SPEED-Bench
low_entropy, temperature 1.0, top-k 20, top-p 0.95, natural EOS, seed 20260822, 512 output cap, one warmup plus three measured repeats. Only steady full-batch pure decode is reported; TTFT, prefill, and endpoint throughput are excluded.The geometric-mean scaling-efficiency gap to FP8 is 2.2% at B8 and 1.0% at B16. At 16K, NVFP4 exceeds the matched FP8 efficiency by 4.4%/6.5% at B8/B16.
Test plan and result
No full-model end-to-end or throughput run was repeated after the final current-main rebase; the audited performance patches are equivalent, the measured optimized instances are unchanged, and the requested merge scope is source plus focused validation.