Genesis vLLM Patches:面向 Ampere GPU 的 int8 端到端推理补丁集
blairq/genesis-vllm-patches · Runtime patches for vLLM 0.29.0: a 27B hybrid-GDN model at 256K context on 2x RTX 3090 — integer attention kernels, W4A8 Marlin, DFlash2 speculative decoding.
该项目是 vLLM 0.29.0 的个人运行时补丁分支,核心思路是在没有 fp8 张量核心的 Ampere(sm_86)硬件上把 int8 作为贯穿各阶段的「计算」格式。补丁覆盖 W4A8 Marlin GEMM、int8 per-token-head KV cache、手写 PTX 整数注意力解码内核、int8 TP all-reduce 以及 int4 lm_head 等,目标是让张量从 RMSNorm 到注意力全程不离开整数域。仓库基于特定机器(2× RTX 3090、TP=2、一个特定量化检查点)调优,附带大量实测对比数据与未解决问题记录,处于活跃开发状态,并非成品。