4.4 NVIDIA H100/H200的HBM架构解析 4.4.1 NVIDIA H100 SXM:HBM3架构标杆 NVIDIA H100(代号Hopper)于2022年发布,是目前最广泛部署的AI训练加速器之一,其HBM子系统设计也是业界HBM3集成的标杆工程。 H100 SXM5版本HBM配置: HBM容量:80GB(5颗HBM3堆叠,每颗16GB) 堆叠配置:每颗HBM3包含12层DRAM die(8层活跃+4层作为缓冲层)+ 1层逻辑层(base die),共13层堆叠 总带宽:3.35TB/s(5颗×672GB/s/颗) 数据速率:4.8Gbps/pin(H100使用的HBM3为初始版本,非最高速率的6.
NVIDIA H100(代号Hopper)于2022年发布,是目前最广泛部署的AI训练加速器之一,其HBM子系统设计也是业界HBM3集成的标杆工程。
H100 SXM5版本HBM配置:
HBM堆叠的物理布局:
H100芯片使用台积电4N(4nm级)工艺制造,面积达814mm²。5颗HBM3堆叠通过硅中介层(interposer)与GPU核心互连。中介层采用台积电CoWoS(Chip-on-Wafer-on-Substrate)封装技术,将GPU和HBM封装在同一有机基板上。
HBM堆叠在GPU核心的四周分布,这种布局有利于缩短GPU到HBM的信号走线长度,降低延迟和功耗。GPU核心内部包含多个HBM控制器,每个控制器管理若干HBM通道,与HBM堆叠的通道一一对应。
HBM控制器设计是H100内存子系统的核心。H100集成了一组高效的HBM内存控制器,具备以下特性:
HBM到SM的数据路径:
在H100中,从HBM读取的数据需要经过以下路径:
HBM堆叠 → TSV/微凸点 → 硅中介层 → GPU封装 → HBM控制器 → L2 Cache → SM
H100的L2缓存容量为40MB,分布在所有SM之间共享。L2缓存充当HBM和SM之间的缓冲层,过滤重复访问、合并写操作。L2缓存的带宽可达3-4TB/s,与HBM带宽匹配,确保不会成为额外瓶颈。
NVIDIA在H100中引入了多项针对HBM带宽优化的技术:
Tensor Memory Accelerator(TMA):
TMA是H100中专门用于加速张量数据搬运的硬件单元。传统架构中,数据搬运需要SM消耗宝贵的计算周期来执行LD/ST指令。TMA将这些任务卸载到专用硬件,SM只需发出一个异步搬运请求即可继续执行计算。这实现了计算与访存的重叠(overlap),显著提升了HBM带宽的有效利用率。
在大型语言模型训练中,TMA可以将HBM的带宽利用率从约60-70%提升到80%以上。
异步拷贝与多级缓存协同:
H100支持以下异步操作模式:
稀疏注意力优化:
针对Transformer架构中的注意力计算,H100支持稀疏注意力模式,通过选择性计算注意力矩阵的非零元素来减少HBM访问量。配合FlashAttention算法(一种HBM友好的注意力实现),可将注意力计算的HBM访问量减少约2-4倍。
NVIDIA H200(代号Hopper H200)于2024年初发布,是首批大规模部署HBM3E的产品之一。
H200 SXM版本HBM配置:
H200 HBM升级的关键意义:
H200与H100的架构差异:
H200在GPU核心层面与H100基本相同(同为Hopper架构),主要区别在于:
这种"核心不变、内存升级"的策略降低了设计风险和开发成本,体现了NVIDIA产品规划的精细考量。
LLM训练场景:
以训练175B参数的GPT-3类模型为例,单台H100集群(8×H100 SXM)的HBM带宽利用情况:
LLM推理场景:
推理场景的HBM带宽效率更高,因为计算强度通常低于训练:
科学计算场景:
H100在HPC基准测试中同样表现出色:
NVIDIA对HBM供应链的管理也是业界关注的焦点:
H100/H200的成功不仅是GPU架构的胜利,更是HBM集成工程的胜利。它们为业界展示了如何在芯片层面充分利用HBM的高带宽特性,以及系统级优化对释放HBM性能潜力的关键作用。