拿下技术面上传简历

线上要做 32K 长上下文的多轮对话。模型 72B、80 层、GQA 8 个 KV head、head_dim 128,KV 用 FP16 存。现在发现单条序列的 KV 就要吃掉十几 GB,batch 只能开到 4,吞吐上不去,但显存已经见底。请先把这个账算清楚,再给出至少三条处于不同层次的优化路径,并说明各自的代价。

AI Infra 与 Harness 工程推理与部署5 年以上知识点 1 个

面试官在考什么

能否从第一性原理算出 KV 的真实开销、确认它是主瓶颈,并在系统、精度、算法、业务四个层次给出有代价说明的方案组合。

KV Cache

回答框架 · 前两步

  1. 会算账:per-token KV = 2(K/V) × layers × kv_heads × head_dim × dtype_bytes = 2×80×8×128×2 ≈ 320KB/token,32K 上下文单序列约 10GB,确认 KV 是第一瓶颈而不是权重。
  2. 系统层:PagedAttention 降低碎片让更多序列共存;prefix caching 复用 system prompt 与公共前缀;max_num_seqs 和 block 预算要按显存反推,而不是拍脑袋。

还有 3 条答案要点、 3 层追问、 4 个常见坑, 以及针对你简历的追问路径,在定制报告里。

上传简历,生成我的题库先看示例报告

同一能力域的其他题