跳到正文
GitHub · 灵魂体仓库·· 19 天前精选AI 评分72

LLM-Inference-Engine:在单张 24GB GPU 上从零手写 Qwen3-8B 推理引擎并逐层实测

PradHolla/LLM-Inference-Engine

来源与原文

GitHub · 灵魂体仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/PradHolla/LLM-Inference-Engine

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

作者开源了 LLM-Inference-Engine 项目,在单张 A10G 上从零构建 Qwen3-8B 推理引擎,容量较朴素服务提升 17.2 倍(0.33 到 5.70 req/s),p95 首 token 延迟从 25.1 秒降到 336 毫秒。项目先写预测再实测,约三分之一预测出错;核心发现是每个优化没有固定加速比,同一量化标记在短请求上值 1.07 倍、长请求上值 2.35 倍,prefix cache 在唯一提示上无收益、对话中达 35 倍,并用 Nsight 定位与 vLLM 的 6 毫秒差距主要来自 2,079 次未融合的 kernel 启动。

推荐理由

作者把每个优化都给出预测、实测和误差解释,量化或缓存的收益取决于工作负载这一点有对照数据支撑,方法可直接迁移到自己的推理项目。

来源:GitHub · 灵魂体仓库 · github.com