#228·nano-vllm

[change] Int8 KV Cache + Async Pipeline + Head-major reordering for 22% Throughput Boost

Author: naalo2Created May 8, 2026Updated May 8, 2026

I'd like to share a downstream project built on top of Nano-vLLM that optimizes KV cache IO patterns and overlaps KV cache writeback with attention computation via async pipelining: Nano-vLLM-kv-compression.

Repo: https://github.com/naalo2/nano-vLLM-kv-compression

new features include:

  • Int8 KV Cache Compression — 50% memory reduction via dynamic per-head quantization
  • Coalesced Layout — Head-major reordering for warp-level memory coalescing
  • GQA-Optimized Flash Attention — group Q-head CTA mapping eliminates redundant KV loads
  • Async KV Store Pipeline — Multi-stream architecture overlaps KV quantization and cache writeback with attention computation

Performance

Setup: RTX 3090, Qwen3-0.6B, 256 requests, random 100–1024 input/output tokens

Engine Time (s) Throughput (tok/s) Speedup
Nano-vLLM 33.05 4,052.56
Nano-vLLM-kv-compression 27.00 4,962.21 +22.4%

Source: GeeeekExplorer/nano-vllm