Minho Ha, Euiseok Kim, Hoshik Kim
SK hynix Inc.
LCA 2026
Presenter: wzw
Date: 2026-05-24
HBM 带宽高但容量太小,CAG/shared pre-computed KV cache 一上到 1M/10M 上下文,就会把系统逼成“多堆 GPU 才能存下”的成本问题。H^3 = HBM + HBF,把巨型只读的模型权重和 shared KV 放进 HBF,把生成期 KV cache 留在 HBM,再用 Latency Hiding Buffer (LHB) 预取遮住 HBF 的 us 级延迟。Llama 3.1 405B 的模拟评估中,10M 场景最大 batch size 提升到 18.8x,throughput 提升 6.14x,throughput per power 最高 2.69x。Cache-Augmented Generation (CAG) 这类巨型只读 shared KV cache 的场景。KV cache;推理时主要是反复读取这份 cache,而不是频繁改写它。HBF 的长处: 大容量 + 高带宽读取,同时避开其短板: 高写延迟 / 差写耐久。
Llama 3.1 405B 举例: shared pre-computed KV cache 在 1M 和 10M 序列长度下约为 540 GB 和 5.4 TB。
HBF 定义成一种 TSV 叠层 NAND flash,目标是提供接近 HBM 的带宽与约 16x HBM 的容量。ns 上升到 us,写耐久更差,单位功耗最高可到 HBM 的 4x。HBF 不能单独当通用主存,它需要一个只读、带宽敏感、且访问顺序可预测的 workload。CAG / gigantic shared KV cache 正好满足这个 workload 画像: 数据巨大、几乎只读、访问顺序可预测、且容量可以直接换 batch。
HBM 直连 GPU shoreline,HBF 通过 HBM base die 级联,二者一起暴露为 GPU 的 main memory。address decoder & router + LHB,所以这不是 PCIe 外挂 Flash,而是近封装异构主存。
model weights + shared pre-computed KV cache,因为它们巨大、只读、带宽敏感。generated KV cache 和其他需要低延迟、频繁写入的数据。HBF 负责“存得下”,HBM 负责“写得动”。HBF 中的模型权重与 shared pre-computed KV,同时从 HBM 读取当前生成阶段的 KV。KV cache 写回 HBM,而不是写去 HBF。prefetch hint。LHB 在后台把后续层需要的 HBF 数据提前拉近,于是 HBF 的长延迟被尽量藏在 layer 间隙里。HBF 最大的问题不是带宽,而是 us 级访问延迟,直接裸读会把 GPU 卡死。Latency Hiding Buffer (LHB),本质上是 base die 上的 tensor-level prefetch buffer。$$ \mathrm{Capacity}{LHB} = 2 \times \mathrm{BW} $$} \times \mathrm{Latency}_{HBF
40 MB SRAM。3 nm SRAM density = 0.021 um^2/bit 估算,SRAM core 面积约 6.72 mm^2;算上 20% 额外电路后总计 8.06 mm^2。121 mm^2 base die 的 6.7%,作者认为可接受。HBF bandwidth ~= HBM bandwidth 的假设上,真实产品化时这里最不确定。Llama 3.1 405B,并把 context 从原始 128K 扩到假设的 1M 和 10M。NVIDIA B200;HBM3e = 192 GB, 8 TB/s;HBF = 3 TB, 8 TB/s。1M 评估用 8 GPUs,10M 用 32 GPUs,这些都是为 HBM-only case 能跑起来设定的最小规模。
HBM-only,红色是 H^3。1M 场景下,8 GPU 时最大 batch size 约提高到 2.6x;10M 场景最高达到 18.8x。
normalized TPS per batch,右图是 normalized TPS per batch per power。1M 场景 throughput 提升 1.25x;10M 场景提升到 6.14x。HBF 更耗电,H3 的 throughput per power 最高仍达到 2.69x,说明容量收益没有被功耗吞掉。H^3 (1/2 bandwidth HBF),也就是把 HBF 带宽直接砍半。1M 场景 throughput 仍不差于 HBM-only;10M 场景 throughput per power 仍有 2.09x。HBF 真能接近文中目标规格,那么 HBM + HBF 对 gigantic read-only KV 场景确实有很强的系统收益。HBF 器件能否同时满足带宽、功耗、热设计与控制器复杂度要求。更大容量 -> 更大 batch -> 更高 throughput -> 更好 throughput per power。H^3,把 HBM 和 HBF 组织成统一编址的近封装异构主存。HBF 的场景: 巨型只读模型权重 + shared pre-computed KV cache。10M case 的最大 batch size 提升 18.8x,throughput 提升 6.14x,throughput per power 最高 2.69x。HBF 概念,而是明确给出最合适的数据形态和一个可算账的 LHB。HBF 规格本身就是目标值,不是实测值,离真实产品还有明显距离。HBF latency 敏感性、真实 runtime prefetch hint、以及和 CXL / host-side tiering 的正面对比。framework hint + LHB + heterogeneous placement 能否在真实软件栈里协同工作。