Skip to content

vLLM Reference Audit

This audit maps the local vLLM-related reference work to turboquant. Each later vLLM phase must update the affected row before the phase is treated as done.

Status values:

  • present: implemented and tested in this repository.
  • partial: some behavior exists, but a listed runtime path or test group is absent.
  • missing: no implementation exists here yet.
  • rejected: not being adopted, with a reason.

File Map

Reference file Feature name Data structures Public functions/classes Runtime path Tests Our target Status
turboquant-vllm/src/turboquant_vllm/lloyd_max.py Lloyd-Max codebooks codebook centroids, boundaries solve_lloyd_max, LloydMaxCodebook quantizer setup test_lloyd_max.py turboquant/core/codebook.py present
turboquant-vllm/src/turboquant_vllm/quantizer.py MSE and QJL codecs indices, norms, signs TurboQuantMSE, TurboQuantProd HF and TQ4 diagnostics test_quantizer.py turboquant/core/ present
turboquant-vllm/src/turboquant_vllm/compressors.py key/value compressors compressed keys, compressed values key and value compressor classes diagnostic compression test_compressors.py turboquant/kv/keys.py, turboquant/kv/values.py partial
turboquant-vllm/src/turboquant_vllm/kv_cache.py HF dynamic cache wrapper compressed layers, lifecycle state dynamic cache wrapper, stats methods HF diagnostic path test_kv_cache_*.py turboquant/integration/hf/dynamic_cache.py partial
turboquant-vllm/src/turboquant_vllm/verify.py model verification model shape record, per-layer metrics config detector, verification runner CLI quality check test_verify*.py turboquant/integration/vllm/verify.py partial
turboquant-vllm/src/turboquant_vllm/benchmark.py A/B measurement run result dicts model load, patched run, benchmark runner local benchmark test_benchmark.py scripts/qwen_ab_eval.py partial
turboquant-vllm/src/turboquant_vllm/triton/tq4_compress.py TQ4 pack kernel packed index bytes, fp32 norms tq4_compress packed cache write test_triton_multi_dim.py turboquant/triton/tq4_update.py partial
turboquant-vllm/src/turboquant_vllm/triton/tq4_decompress.py TQ4 unpack kernel packed index bytes, norms tq4_decompress diagnostic readback test_tq4_out_parameter.py turboquant/triton/tq4_decode.py partial
turboquant-vllm/src/turboquant_vllm/triton/fused_qk_attention.py packed score kernel packed key rows fused_qk_scores score-only diagnostic kernel tests turboquant/triton/tq4_decode.py missing
turboquant-vllm/src/turboquant_vllm/triton/fused_paged_tq4_attention.py TQ4 paged decode block table, packed pages fused_paged_tq4_decode decode attention test_fused_paged_tq4_attention.py turboquant/triton/tq4_decode.py partial
turboquant-vllm/src/turboquant_vllm/triton/fused_paged_tq4_int8_prefill.py INT8-Q prefill int8 query scale, packed pages prefill wrapper prefill experiment test_fused_paged_tq4_int8.py turboquant/triton/int8_prefill.py missing
turboquant-vllm/src/turboquant_vllm/vllm/tq4_backend.py plugin-style backend TQ4 spec, backend, impl backend registration, page shape, dispatch vLLM plugin path test_vllm_*.py turboquant/integration/vllm/ partial
vllm-turboquant/vllm/v1/attention/ops/turboquant_kv_cache.py recipe layout and vector codec group layout, kernel meta layout, transforms, pack/unpack native vLLM recipe path test_turboquant.py turboquant/integration/vllm/recipe.py, vector.py partial
vllm-turboquant/vllm/v1/attention/ops/turboquant_metadata.py metadata JSON tensor/layer/calibration metadata load, save, discovery, TP slicing native vLLM recipe path test_turboquant.py turboquant/integration/vllm/metadata.py partial
vllm-turboquant/vllm/v1/attention/ops/triton_turboquant_kv_update.py recipe update kernel packed recipe page rows turboquant_write_packed_kv KV cache update test_turboquant.py turboquant/triton/recipe_update.py missing
vllm-turboquant/vllm/v1/attention/ops/triton_turboquant_decode.py recipe decode kernel block table, token metadata, packed pages decode forward wrapper paged attention test_turboquant.py turboquant/triton/recipe_decode.py missing
vllm-turboquant/vllm/v1/attention/backends/triton_attn.py backend integration metadata builder, layer impl state update, decode, fallback gates vLLM runtime test_turboquant.py turboquant/integration/vllm/runtime.py missing
vllm-turboquant/vllm/config/cache.py cache config gate cache dtype, gate, metadata path config validators vLLM config config tests turboquant/integration/vllm/registry.py missing
vllm-turboquant/vllm/engine/arg_utils.py CLI flags engine args enable flag, metadata path vLLM CLI config tests scripts/verify_vllm_integration.py missing
vllm-turboquant/benchmarks/generate_turboquant_metadata.py activation calibration activation accumulator, metadata builder metadata generator CLI recipe metadata build test_turboquant.py turboquant/integration/vllm/calibration.py, scripts/generate_vllm_metadata.py missing

Feature Checklist

Feature group Our target Status Next action
Lloyd-Max codebooks turboquant/core/codebook.py present keep exact Beta tests
QR/RHT/block-Hadamard transforms turboquant/core/rotation.py, turboquant/integration/vllm/vector.py partial expose query/group helpers
QJL residual turboquant/core/qjl.py, turboquant/integration/vllm/vector.py partial add recipe decode contract
bit packing turboquant/packing/bits.py present keep 1/2/3/4/5-bit tests
TQ4 cache wrapper turboquant/integration/hf/dynamic_cache.py partial add Transformers model tests
TQ4 packed page layout turboquant/integration/vllm/page_cache.py present add kernel-backed writer
asymmetric K/V bits turboquant/integration/vllm/recipe.py, page cache partial add page bytes and tests
turboquant25/35 recipe layout turboquant/integration/vllm/recipe.py partial add public aliases from plan
metadata JSON turboquant/integration/vllm/metadata.py partial validate recipe and runtime shape
activation calibration turboquant/integration/vllm/calibration.py present add model-family tests
TP metadata slicing turboquant/integration/vllm/metadata.py present expand tests
vLLM config hooks turboquant/integration/vllm/registry.py missing add gate checks
vLLM cache dtype registry turboquant/integration/vllm/registry.py missing map dtype to uint8
vLLM attention backend selection turboquant/integration/vllm/runtime.py missing add dispatch layer
packed KV update kernel turboquant/triton/tq4_update.py, recipe_update.py missing add reference then Triton
packed paged decode kernel turboquant/triton/tq4_decode.py, recipe_decode.py missing add reference then Triton
INT8-Q prefill turboquant/triton/int8_prefill.py missing add gated experiment
CUDA graph buffers turboquant/integration/vllm/runtime.py missing add buffer owner
SWA/global-layer behavior HF and vLLM adapters missing add config tests
model verification turboquant/integration/vllm/verify.py partial add real model smoke
Qwen A/B benchmark scripts/qwen_ab_eval.py partial add vLLM variants