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% 读流量。

方案

  1. 共封装层次:HBM 上叠 HBF;GPU 直访 HBM,HBF 仅经 D2D + base-die 缓冲(预取/驱逐双缓冲)。
  2. 放置:活跃 decode KV → HBM;暂停冷池 → HBF;resume 前预取回 HBM。
  3. 写策略:write-on-evict(块最多写一次),对比 all-KV-to-flash。
  4. 评测设定:B200 192 GiB HBM + 3 TiB SLC HBF(读带宽对齐 HBM);agentic 轨迹放大并发。

效果(仅论文数字)

指标数字
TBT @ Na=48 / 12814 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)
并发会话 / GPU24×
读功耗 vs all-flash20 pJ/bit 下每卡 ≈950 W,8-GPU 7.6 kW(能量扫描 2.1–12.2 kW/节点)
寿命 @ Na=48write-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 — 封装内热冷路径

Citations

[1] arXiv PDF — Baek et al., arXiv:2609.25782 [2] raw stub