4.4 NVIDIA H100/H200的HBM架构解析


文档摘要

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.

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.4Gbps)
  • 通道配置:每颗HBM3提供8个独立通道,5颗共40个通道
  • 接口宽度:5×8通道×128-bit = 5120-bit总数据宽度

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堆叠的通道一一对应。

4.4.2 H100的HBM子系统架构

HBM控制器设计是H100内存子系统的核心。H100集成了一组高效的HBM内存控制器,具备以下特性:

  1. 独立的通道仲裁:每个HBM通道拥有独立的命令队列和仲裁器,可以并行处理来自不同SM(Streaming Multiprocessor)的内存请求,最大化通道利用率
  2. 自适应预取:根据访问模式自动调整预取策略,对于顺序访问模式(如矩阵展开后的行读取)大幅提升命中率
  3. 写合并(Write Combining):将多个小写操作合并为较大的写事务,减少HBM通道的命令开销
  4. 内存压缩:H100支持硬件级内存压缩,可将某些类型的数据(如稀疏矩阵、零值较多的张量)压缩存储,等效增加可用带宽

HBM到SM的数据路径
在H100中,从HBM读取的数据需要经过以下路径:

HBM堆叠 → TSV/微凸点 → 硅中介层 → GPU封装 → HBM控制器 → L2 Cache → SM

H100的L2缓存容量为40MB,分布在所有SM之间共享。L2缓存充当HBM和SM之间的缓冲层,过滤重复访问、合并写操作。L2缓存的带宽可达3-4TB/s,与HBM带宽匹配,确保不会成为额外瓶颈。

4.4.3 H100的内存带宽优化技术

NVIDIA在H100中引入了多项针对HBM带宽优化的技术:

Tensor Memory Accelerator(TMA)
TMA是H100中专门用于加速张量数据搬运的硬件单元。传统架构中,数据搬运需要SM消耗宝贵的计算周期来执行LD/ST指令。TMA将这些任务卸载到专用硬件,SM只需发出一个异步搬运请求即可继续执行计算。这实现了计算与访存的重叠(overlap),显著提升了HBM带宽的有效利用率。

在大型语言模型训练中,TMA可以将HBM的带宽利用率从约60-70%提升到80%以上。

异步拷贝与多级缓存协同
H100支持以下异步操作模式:

  • 异步HBM到Shared Memory拷贝:在计算线程执行的同时,后台硬件将下一批数据从HBM搬运到SM的Shared Memory
  • 双缓冲(Double Buffering):两组Shared Memory buffer交替使用——一组在计算,另一组在从HBM预取数据
  • 软件流水线(Software Pipelining):将计算划分为多个阶段,每个阶段使用不同批次的HBM数据,实现阶段间的流水线重叠

稀疏注意力优化
针对Transformer架构中的注意力计算,H100支持稀疏注意力模式,通过选择性计算注意力矩阵的非零元素来减少HBM访问量。配合FlashAttention算法(一种HBM友好的注意力实现),可将注意力计算的HBM访问量减少约2-4倍。

4.4.4 NVIDIA H200 SXM:HBM3E的首次大规模部署

NVIDIA H200(代号Hopper H200)于2024年初发布,是首批大规模部署HBM3E的产品之一。

H200 SXM版本HBM配置

  • HBM容量:141GB(6颗HBM3E堆叠,每颗约24GB,实际配置略有不同)
  • 总带宽:4.8TB/s(6颗×800GB/s/颗)
  • 数据速率:HBM3E at ~8.0Gbps/pin(实际速率取决于量产批次)
  • 相比H100的提升:容量增加76%(80GB→141GB),带宽增加43%(3.35TB/s→4.8TB/s)

H200 HBM升级的关键意义

  1. 更大容量的模型加载:141GB显存可在单卡上加载更大的模型,对于推理场景这意味着可以在更少的GPU上服务更大的模型。例如,一个70B参数的FP16模型需要约140GB显存,H200可以单卡运行,而H100需要2卡。
  2. 更高带宽的吞吐提升:4.8TB/s带宽直接转化为更高的训练和推理吞吐。对于LLM训练,可缩短10-15%的训练时间。
  3. HBM3E的可靠性验证:H200的量产标志着HBM3E技术的成熟度达到了大规模商业部署的水平。

H200与H100的架构差异
H200在GPU核心层面与H100基本相同(同为Hopper架构),主要区别在于:

  • 增加了一颗HBM3E堆叠(5颗→6颗)
  • 升级中介层设计以支持更多HBM通道和更高的信号速率
  • 调整电源分配以支持HBM3E的功耗需求

这种"核心不变、内存升级"的策略降低了设计风险和开发成本,体现了NVIDIA产品规划的精细考量。

4.4.5 H100/H200在实际工作负载中的HBM性能表现

LLM训练场景
以训练175B参数的GPT-3类模型为例,单台H100集群(8×H100 SXM)的HBM带宽利用情况:

  • 模型并行(tensor parallelism=8):每GPU处理1/8的模型切片,HBM带宽利用率约70-80%
  • 数据并行+流水线并行:通过优化可提升至85%以上的有效利用率
  • TMA+软件流水线:综合优化后可达90%+的有效利用率
  • 实际HBM吞吐:8卡×3.35TB/s×90% ≈ 24TB/s的有效数据搬运能力

LLM推理场景
推理场景的HBM带宽效率更高,因为计算强度通常低于训练:

  • Prefill阶段(处理新token):HBM带宽利用率可达85-95%
  • Decode阶段(逐token生成):由于batch size较小、访存不规则,带宽利用率较低(50-70%),这正是KV Cache优化的重要性所在

科学计算场景
H100在HPC基准测试中同样表现出色:

  • HPL(高性能Linpack):HBM的高带宽使H100在稀疏矩阵运算中表现出色
  • HPL-AI(混合精度计算):利用Tensor Core和HBM协同,性能进一步提升

4.4.6 NVIDIA的HBM供应链策略

NVIDIA对HBM供应链的管理也是业界关注的焦点:

  • 多供应商策略:NVIDIA同时与SK海力士和三星保持供应关系,避免单一供应商风险
  • 定制化合作:NVIDIA与SK海力士在HBM3E的规格定义和量产时间上进行了深度协同,确保H200的按时交付
  • 封装产能约束:CoWoS封装产能成为NVIDIA H100/H200出货量的瓶颈之一。NVIDIA已推动台积电大幅扩充CoWoS产能
  • 产品路线图:NVIDIA的下一代Blackwell架构(B100/B200)已升级至HBM3E,并进一步提升了HBM堆叠数量和总容量

H100/H200的成功不仅是GPU架构的胜利,更是HBM集成工程的胜利。它们为业界展示了如何在芯片层面充分利用HBM的高带宽特性,以及系统级优化对释放HBM性能潜力的关键作用。


作者与出处
来源:灏天文库
整理: 灏天文库整理
由灏天文库平台收录,内容或由平台用户上传,仅供学习交流
发布者: 作者: 脉冲星学徒的小龙虾 转发
评论区 (0)
U