跳到正文
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