AMD Victor Peng:AI对电力的渴求凸显3D堆叠封装的重要性

作者:Julian

摩尔定律可能已经显现疲态,但仍有调整空间和操作余地

在本周召开的Hot Chips大会上,AMD总裁Victor Peng探讨了半导体行业面临的一个重大挑战,即应对日益增长的AI模型需求所带来的电力问题。

他指出:“如果从宏观角度来看,对于那些大规模的部署,我们甚至可能面临无法找到足够的电力资源,以及对电网和配电的担忧。”

过去一年中,AI对电力的近乎无止境的需求引起了广泛关注,以至于有些运营商开始在核电站旁边设立数据中心。并且,这个问题不会变得更容易解决。

他表示:“事实证明……如果你增加计算能力,扩大模型规模,你将获得更好的性能、精度和智能水平,不管你如何看待这些。”他指出,这些模型的电力需求已迅速从数百兆瓦时增加到数百吉瓦时。

为应对这一挑战,Victor认为,半导体行业需要更加关注如何使基础设施不仅更高效而且性能更强。

他解释说:“无论你认为你的电力预算有多大限制,如果你获得更高的性能,你可以更快地训练更大规模的模型,或者以更经济的方式提供服务。”

不幸的是,芯片制造商和设计者们过去依赖的许多调整手段已经难以为继。工艺技术的改进变得越来越小,而每一代之间的时间变得更长,成本也越来越高。

Victor认为,业界认为摩尔定律依然有效的原因在于,许多这些挑战可以通过转向芯粒架构和先进封装技术来克服。这是AMD一直引领的领域,可以追溯到2017年其首代Epyc处理器的发布。

1.jpg

根据 AMD 的数据,3D 堆叠硅片虽然更复杂,但其效率是分立方案的 50 倍。

通过采用3D堆叠硅片技术——如我们在AMD的X系列Epyc处理器、Ryzen处理器以及MI300系列GPU和APU中所见——Victor认为,与分立方案相比,这可以实现每焦耳能源消耗下50倍的更高位处理能力。

当你开始考虑扩展计算规模时,这一点变得尤为重要——这是当今数据中心中AI训练和大型模型推理中非常常见的需求。

根据AMD的数据,与将一切保持在芯片上相比,扩展计算系统——如基于NVLink或Infinity Fabric的GPU网络——需要1600倍的功率。而要跨多个节点扩展计算,还需要更多的电力,这部分电力主要消耗在较慢的互连设备的低效性,以及运行所有交换机、网络接口卡(NIC)和光纤所需的电力上。

数据中心.jpg

即使使用像Infinity Fabric这样快速的互连结构,3D堆叠封装芯片的成本依然非常巨大。

Victor表示,网络仍然是提高数据中心效率的一个机会。虽然计算占据了电力消耗的大部分,但网络大约占据了其中的20%。

在这里,他指出了可能帮助的扩展网络架构,并提到用于连接AMD MI300X系统中的八个GPU的Infinity Fabric。竞争对手Nvidia已经展示了使用NVLink连接多达32个GPU的系统,并计划推出密度更高的36和72个GPU配置。

然而,AI的电力问题并不仅限于数据中心。它还扩展到客户端和嵌入式应用场景中——只不过这里所讨论的不是几千瓦特,而是几十瓦或更少。而且,每个领域除了电力以外,还有不同的需求——如延迟——需要考虑。

在这些领域,Victor认为,异构计算的合理应用提供了一条前进的道路。自AMD在2022年收购Xilinx和Pensando以来,其硬件产品线涵盖了CPU、GPU、DPU、FPGA和NPU。最近,我们开始看到这些技术集成到移动芯片中,使得AI处理的能耗进一步降低。

最新的例子是AMD的Strix Point Ryzen 300系列处理器,配备了XDNA 2 NPU,能够实现50 TOPS的INT8和Block FP16性能。竞争对手Intel、Qualcomm和Apple也出于同样的原因采用了NPU。

• AI/ML的未来取决于当下的现实——而这并不乐观

• 谷歌训练了一个生成AI模型来模拟《毁灭战士》的游戏引擎,几乎接近实时

• 埃隆·马斯克收紧了Grok AI机器人的管理,以阻止选举虚假信息

• 计算机行业的失误——早期的打击

Victor还提到量化在无论是部署AI模型在数据中心还是在边缘计算中都很重要。量化是一种压缩技术,用于将模型权重缩减到较低精度,以换取一些质量损失。

精度损失.png

如果你能接受精度的损失,量化可以带来相当大的效率提升。

AMD已经在MI300X中采用了FP8,并计划明年推出MI350X时支持4位浮点数据类型。正如Victor的主题演讲所强调的那样,为了更高的性能功耗比,这种精度的权衡往往是值得的。

与此同时,在嵌入式领域,Victor建议可以直接将模型映射到硅片上,以优化数据流。在一次内部测试中,AMD的技术人员能够将每次推理的能量消耗减少4500倍,相比标准的INT8计算。

最后,Victor强调了软件优化、协同设计和合作的重要性,以释放硬件的全部性能。在他决定本月底退休之前,Victor在这一领域做出了重大贡献。

最新文章