Hot–Cold Tiering of HBM and High Bandwidth Flash for Agentic LLM Serving
一句话结论
把 agentic KV 的热集留在 HBM、冷池放进共封装 HBF,在 Qwen3-Coder-30B-A3B 上做到 14 ms TBT、resume 仅加 ≈0.1 ms、每 GPU 会话 24×,并相对全 KV 走闪存砍掉约 7.6 kW/8-GPU 读功耗——HBF 是冷层,不是 HBM 替代品。
动机
- Agentic 多轮:tool/用户等待期间会话 idle,但 KV 必须保留;暂停池迅速撑爆 HBM。
- 全 KV 进 HBF:每步 decode 付高读能量 + 写寿命压力(对照 HBFlex 全 HBF、SPLASH 稀疏×HBF)。
- 双峰:Na=48 时热集仅占驻留容量 3%,却贡献 98% 读流量。
方案
- 共封装层次:HBM 上叠 HBF;GPU 直访 HBM,HBF 仅经 D2D + base-die 缓冲(预取/驱逐双缓冲)。
- 放置:活跃 decode KV → HBM;暂停冷池 → HBF;resume 前预取回 HBM。
- 写策略:write-on-evict(块最多写一次),对比 all-KV-to-flash。
- 评测设定:B200 192 GiB HBM + 3 TiB SLC HBF(读带宽对齐 HBM);agentic 轨迹放大并发。
效果(仅论文数字)
| 指标 | 数字 |
|---|---|
| TBT @ Na=48 / 128 | 14 ms(72 tok/s)/ 27 ms(37 tok/s);均 <50 ms SLO |
| Resume 附加 | 相对 prefill ≈0.1 ms(HBF≈0.084 ms;NVLink-C2C 1.0 ms;PCIe 7 ms;recompute 14 ms) |
| 并发会话 / GPU | 24× |
| 读功耗 vs all-flash | 20 pJ/bit 下每卡 ≈950 W,8-GPU 7.6 kW(能量扫描 2.1–12.2 kW/节点) |
| 寿命 @ Na=48 | write-on-evict ≈20 年;all-flash ≈11 年 |
口径: IEEE CAL 风格 trace 仿真,非硅实测 tok/s。
与 wiki 的关系
- HBFlex — 全 HBF 运行时;本文坚持 热集必须 HBM
- SPLASH — 长上下文稀疏×HBF;本文焦点是 agentic idle 冷池
- UNISON — 近存调度谁该留快层;本文给 物理 HBM/HBF 分层
- End-to-End Memory Data Path — 封装内热冷路径