拿下技术面上传简历

单张 A100 80G 上跑一个 13B 的 MHA 模型:40 层、40 个 KV head、head_dim 128、FP16,权重占 26GB。服务 8K 上下文的并发请求。请先算出每个 token 的 KV cache 占用和单条 8K 序列的占用,然后回答:这张卡理论上最多能服务几个 8K 并发请求,为什么监控里看到的实际并发数往往比这个数还低?在不改模型结构、不换卡的前提下,你怎么把并发提上去,按性价比排序。

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

面试官在考什么

判断候选人能不能把 KV cache 的显存账算对,并分清“理论容量”与“实际可服务并发”的差距来源(预分配、碎片、激活与 CUDA graph),进而给出有代价意识的手段排序。

KV Cache

回答框架 · 前两步

  1. 算对数值:KV/token = 2 × 40 层 × 40 head × 128 × 2B = 800KB;8K 序列 = 6.4GB。忘乘 2 或忘乘层数都是硬伤。
  2. 算容量账并解释差距:80 − 26(权重) − 4~6(激活/CUDA graph/框架) ≈ 48GB,约 7~8 条 8K 序列;如果按 max_model_len 预留式分配还会更低,PagedAttention 按需分块是把它拉回来的第一手段。

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

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

同一能力域的其他题