AI时代GPU发展

AIGPU

2016 年,黄仁勋亲自向 OpenAI 赠送了全球第一台 DGX-1,马斯克也在接收现场。

或许就是这个举动,催生了 ChatGPT 的到来。

DGX-1 被英伟达称作为「AI 超级计算机」,当时捐赠的版本价值 12.9 万美元,集成了 8 块 Tesla P100(Pascal 架构)GPU,整个系统的机器学习算力为 170 teraflops(FP16)。

在此之后,英伟达陆续推出了 DGX-2(Volta 架构)、DGX A100 服务器(Ampere 架构),持续引领了 AI 算力的最高水平。

4.25日赠送给 OpenAI 的 DGX H200,是去年 11 月在全球超算大会上推出的最新产品。英伟达在官方博客中曾表示:H200 Tensor Core GPU 具有改变游戏规则的性能和内存功能,可增强生成式 AI 和高性能计算 (HPC) 工作负载。单块 H200 的 FP16 算力是 1979 TFLOPS。

作为首款采用 HBM3e 的 GPU,H200 借助更大更快的内存可加速生成式 AI 和大型语言模型 (LLM) 的运行,同时推进 HPC 工作负载的科学计算。

老黄送显卡,看来正在成为英伟达的惯例。不过还有一个不得不说的故事,就是神经网络刚刚兴起的时候,英伟达曾经「拒绝」过人工智能先驱 Geoffrey Hinton。

2009 年 Hinton 的研究小组在使用英伟达的 CUDA 平台训练神经网络来识别人类语音。他对研究结果的质量感到惊讶,并于当年晚些时候的一次会议上介绍了这些结果。Hinton 联系了英伟达,「我发了一封电子邮件说,听着,我刚刚告诉一千名机器学习研究人员,他们应该去购买英伟达 GPU。你能给我免费寄来一份吗?」结果英伟达拒绝了。

尽管受到冷落,Hinton 仍然鼓励他的学生使用 CUDA,其中包括传奇学者 Alex Krizhevsky。2012 年,Krizhevsky 和他的研究伙伴 Ilya Sutskever 在预算紧张的情况下从亚马逊购买了两张 GeForce 显卡,训练视觉识别神经网络,这才有了一代经典 AlexNet。

如今:GTC大会上发布的Blackwell芯片,它几乎把所有性能表现都刷新到了一个前所未有的级别。所以,它到底怎么做到的?

首先来看一下它的核心指标。

Blackwell B200 GPU核心由大约1600平方毫米的处理器构成,封装内包含两块硅片,上面集成的晶体管数量是2080亿。这两片硅片采用的是台积电(TSMC)的N4P芯片制造技术,相较于制作Hopper架构GPU(如H100)所使用的N4技术,性能提升了6%。

带宽方面,同Hopper一样,B200同样也配备了高带宽内存。

B200采用的是和H200一样的最新款HBM3e内存,总容量达到了192 GB,相较于第二代Hopper芯片H200的141 GB有所提升。此外,内存带宽也从H200的4.8 TB/s提升至8 TB/s。
根据英伟达的介绍,这个新GPU在训练性能上相较前代Hopper H100 GPU提升了四倍,推理性能提升最高可达30倍,而能源效率更是提高了25倍。

而这些只是开始,英伟达接下来开始“叠buff”,通过各种集成产品来把它的性能在实际场景里发挥到极致。

首先是和CPU合体。B200 GPU和英伟达自家的Grace CPU进行合体,组成GB200。它包含一个 Grace CPU 和两个 Blackwell GPU。GB200利用新型低精度数值格式,最高能实现高达11.5 exaflops的人工智能计算能力。说实话,关注硬件这么多年了,exaflops这个词我甚至是现查的翻译软件才知道具体说的是什么。11.5 exaflops代表1后面接18个0的flops,也就是每秒执行11.5百万亿次浮点运算。
真是字面意义上的,下一个量级。

然后是英伟达口中的“超级计算机”——DGX GB200。它由3600个GB200超级芯片模块组合而成,通过Nvlink链接,以形成一个完整的系统解决方案。DGX GB200提供了超大规模的并行计算能力和极高的内存带宽,目标是服务于大规模的AI训练、推理和高性能计算任务。它有点对标谷歌TPU v5,也是一个完整系统,用来装在服务器上提供算力。

这些惊人的数据让人们关注“芯片”本身,但黄仁勋也在发布会一直强调:Blackwell不是芯片本身,它是一个全新的架构。这个如日中天的技术公司在自己技术上真正核心的迭代改变,其实藏在这些数据背后。
其中两个改进值得关注。其一是它的处理浮点数矩阵运算的单元,其二是它的封装尺寸。

B200的包含了能够处理仅4比特宽度浮点数矩阵运算的单元,这是个啥概念?我们经常听那些研究大模型的人讲一个概念叫做“并行计算”,其本质是说模型中的权重和激活值通常是以浮点数的形式存储的,而矩阵运算(如卷积、矩阵乘法等)是这些模型训练和推理过程中的基本运算。

GPU内置了大量的浮点运算单元,能够同时处理多个浮点数的运算,极大地加速了大规模矩阵运算的速度。这也是大模型需要GPU产品的核心原因。人工智能的计算对精度有特定要求,但并非所有操作都需要最高的精度。GPU支持多种浮点数格式(如FP32、FP16、BFLOAT16甚至INT8等),不同的浮点数格式对应不同的精度和存储需求。

越小的浮点数占用的存储空间更小,计算时所需要的读写次数相应减少,运算速度得到提升。同时对于GPU等并行计算硬件,同一时间内处理的浮点数越多,总的计算吞吐量就越高。较小的浮点数格式允许在同样的芯片面积上集成更多的计算单元,实现更密集的并行计算,从而提升大模型训练和推理的速度。Hopper 可以使用小至 8 位的浮点数格式,Blackwell则是4位,而这种精细度上的变化,是Blackwell的一个“杀招”。

接下来说第二点,简单说,就是“大”。

看了发布会的人可能都会觉得,Blackwell最直观的感受就是大,下图是Blackwell(左)和上一代GPU Hopper(右)的对比,可以明显看出来Blackwell大了整整一圈。

Blackwell(左)Hopper(右)
Blackwell(左)Hopper(右)
我们知道,芯片的封装尺寸越小,代表它的引线更短,需要的寄生电感、电容也就越小,那就有助于降低信号传输时的功率损耗,从而提高能效。同时基础物理知识可以知道,当速度不变时,两地之间距离越近,到达时间也就越快,而更小的封装尺寸意味着信号传播距离更短,能够减少信号延迟,提高速度,这对高速信号传输和高性能处理器尤其重要。那为什么还要往大做呢?

首先小尺寸自己也有问题,最主要就是散热。热量从高温区域沿着温差梯度通过固体传递到低温区域,表面积越小,与周围环境接触的边界越少,热量传递通道就越少,散热效果相对较差。

任何一个拥有服务器集群的公司,他的头号敌人就是热。GPU工作时间既取决于模型训练的具体情况,也受制于基础设施条件和资源管理策略。对于高强度训练场景,GPU保持7*24小时连轴转已经是屡见不鲜了。如此的工作量就会产生大量的热量,这就是为什么越大的服务器集群,越需要高性能的制冷措施。

热对于设备的损耗有哪些呢?最重要的一点是当GPU温度持续高于推荐的工作温度范围时,高温会使内部电子元器件加速老化,甚至导致永久性损坏。例如,硅基半导体在高温下可能出现电子迁移现象,导致电路性能下降,严重时会造成GPU失效。另外,大多数现代GPU都有温度保护机制,一旦温度过高,GPU会自动降低工作频率以减少热量产生,或者触发系统保护措施,直接停止工作以防止硬件损坏。这种情况下,计算机性能会下降,游戏帧率降低,或者导致程序意外终止。

当然了,把封装尺寸做大除了散热效率增加,那里面装的东西肯定也增加了。就像开头咱说的,他里面有2080亿个晶体管,而H100才800多亿个,翻了两倍不止,最大原因就是这回物理意义上的“地方大了”。Blackwell之所以这么大个头,很可能是使用了多芯微(multi chiplet)设计。

这是一种把原本单个大型芯片分割为多个较小的、具有特定功能芯粒(Chiplets)的技术。这些芯粒各自独立制造,然后通过先进的封装技术将其堆叠或并排集成在一起,从而组成一个完整的高性能GPU。这种设计的优点在于小尺寸芯粒的制造良率通常比大尺寸单芯片更高,通过组合多个良品芯粒可以提高最终产品的良率。同时它还允许不同芯粒采用不同的工艺节点进行制造,分别优化性能和能效。最重要的是它芯片里面的空间大了,可以集成更多的晶体管和功能模块,提高整体性能。
这并不是一个新鲜的技术,其他厂商早在尝试,根本原因是以“把芯片做的越来越小”为代表的路线正在走到尽头。比如GH100其实就已经接近当前最先进的4纳米掩模版的上限了。但英伟达这一次还是“后来居上”,因为它展示出来的两个芯片之间的连接速度实在惊人。

黄仁勋称,这两个芯片可以像一个芯片那样,甚至两者自己都并不会区分自己是哪一块。而且,在软件层面,它们会被作为一个统一的CUDA GPU来运行。两者之间的互联速度达到了10Tbps,这比现在已知的Chiplet类的方案都要高。

这种强大的连接能力甚至让英伟达干脆不再使用更先进的工艺,而是继续在4纳米工艺上做迭代。

黄仁勋并没有详细介绍英伟达是如何做到的。他只是在演讲里淡淡的装了一把:

“当我们被告知Blackwell的野心超出了物理学的极限时,我的工程师说,那又怎样?”

总结:

这就是中国工程师和国外工程师的差距,为什么国外可以做工程师做到50-60-70退休,而国内很多工程师都到30多岁。

  • 11
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值