第 2026-10-18 期 · 行业追踪 · 研究 · 开源
从零实现GPT-2 KV缓存并测量性能
一位开发者用PyTorch从零实现了GPT-2 small的KV缓存,并测量了延迟和内存。在上下文长度1000时,缓存将每token时间从29.69毫秒降至4.95毫秒,生成1016个token的总时间从14.66秒降至5.03秒。缓存与未缓存的贪心生成结果在200个token内一致。
查看原文 ↗一位开发者用PyTorch从零实现了GPT-2 small的KV缓存,并测量了延迟和内存。在上下文长度1000时,缓存将每token时间从29.69毫秒降至4.95毫秒,生成1016个token的总时间从14.66秒降至5.03秒。缓存与未缓存的贪心生成结果在200个token内一致。
查看原文 ↗