GitHub · 去对齐工具与仓库·· 18 天前精选AI 评分56
mini-infer-engine:从零实现的 LLM 推理引擎,在 CPU 上复现 vLLM 核心机制
pandeylakshya207-max/mini-infer-engine
来源与原文
GitHub · 去对齐工具与仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/pandeylakshya207-max/mini-infer-engine
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
作者开源了从零编写的 LLM 推理引擎 mini-infer-engine,基于 Qwen2.5-0.5B-Instruct 在纯 CPU 笔记本上完整实现四个阶段并实测。连续批处理把 8 请求工作负载吞吐从 11.55 tok/s 提到 35.61 tok/s(3.08x);Paged KV-cache 因未写定制 fused kernel 且测试提示长度相近,吞吐降至 13.94 tok/s、内存节省 0%,作者如实记录这是工作负载相关的取舍;优先级调度器通过抢占让 HIGH 请求实现 0ms 等待、98ms 首 token 延迟,被抢占请求总耗时从 8.06s 增至 10.49s。
推荐理由
作者从零实现了连续批处理、Paged KV-cache 和优先级调度,并用同负载实测各阶段吞吐,可作推理系统学习参考。
来源:GitHub · 去对齐工具与仓库 · github.com