👉 腾小云导读
近期大量 AIGC 产品横空出世,可以聊天、写代码、解答难题、写小说,饱受热捧。其技术基座大模型的给力支持,往往伴随着大规模、长时间的 GPU 集群训练任务。这对网络互联底座的性能、可靠性、成本等各方面都提出极致要求。业界主流 GPU 集群网络技术路线是什么?腾讯的解决方案是什么?腾讯工程师何春志将带来最新解读。欢迎阅读。
👉 看目录,点收藏
1 业界主流 GPU 集群网络技术路线
2 如何创造AI训练集群下的极致性能网络
2.1 超带宽计算节点
2.2 多轨道流量聚合架构
2.3 异构网络自适应通信
2.4 定制加速通信库
3 驾驭高性能——最大以太 RDMA 网络的工程实践
3.1 端网部署一体化
3.2 全栈巡检,一键故障定位
3.3 业务无感秒级网络自愈
4 总结
AI大模型以其优异的自然语言理解能力、跨媒体处理能力以及逐步走向通用AI的潜力成为近年AI领域的热门方向。业内头部厂商近期推出的大模型的参数量规模都达到了万亿、10万亿级别。GPT-3大模型参数量多达1750亿个,而刚刚发布的GPT-4大模型更是赋予了升级版多模态任务处理能力,在多个任务领域取得重大进步。
据报道,GPT-3 的训练使用了微软专门建设的 AI 计算系统,由1万个 V100 GPU 组成的高性能网络集群,总算力消耗约 3640 PF-days (即假如每秒计算一千万亿次,需要计算3640天)。如此大规模、长时间的 GPU 集群训练任务,对网络互联底座的性能、可靠性、成本等各方面都提出极致要求。
01、业界主流 GPU 集群网络技术路线
GPU 计算集群的高性能网络建设并非易事,需求综合考虑多种因素。比如网络规模、网络拓扑架构选型、接入带宽与网络容量、网络协议选择、甚至与计算芯片的联合优化设计等等。为了解决上述关键问题,业界厂商通常会根据自身需求,选择不同的网络技术路线。
总体来说业界呈现出3大网络技术路线趋势:
-
商用网络方案:采用商用 GPU+ 商用网络组网、协议,以 Nvidia 售卖的 DGX SuperPod 为代表。该方案整体集成度高,网络深度优化,性能最优,但是价格高昂。
-
自研以太网络方案:采用商用 GPU+ 自研网络协议,该方案经过自研协议优化后,性能接近商用方案,成本低。
-
自研计算芯片+自研网络方案:采用自研计算芯片+自研网络协议,能够做到计算芯片与网络深度配合优化,性价比最优。
以上技术路线都会对 GPU 集群网络做深度定制,通过追求极致网络性能,来支撑集群极致算力。
02、如何创造AI训练集群下的极致性能网络
面向AI大模型训练需求,腾讯推出了业界领先的高性能计算网络架构——星脉。 星脉网络在极致高性能上,采用 1.6T 超带宽接入、多轨道聚合流量网络架构、异构网络自适应通信优化技术、定制加速通信库,构建了 1.6T ETH RDMA 网络,实现了 AI 大模型通信性能的10倍提升、GPU 利用率40%提升、通信时延降低 40%,单集群规模达到 4K(最大规模64K)。基于全自研网络硬件平台网络建设成本降低 30%,模型训练成本节省30%~60%。
在高可用保障上,通过全自动化部署配置核查,覆盖服务器 NUMA、PCIE、NVSwitch、NCCL、网卡、交换机数百个配置项,并通过实时 Service Telemetry 技术监控业务系统运行效率,保障大规模集群部署,实现性能实时监控与故障告警。
图1. 星脉高性能计算网络
面对千亿、万亿参数规模的大模型训练,仅仅是单次计算迭代内梯度同步需要的通信量就达到了百 GB 量级。此外还有各种并行模式、加速框架引入的通信需求,使得传统低速网络的带宽远远无法支撑 GPU 集群的高效计算。因此要充分发挥 GPU 计算资源的强大算力,必须构建一个全新的高性能网络底座,用高速网络的大带宽来助推整个集群计算的高效率。
2.1 超带宽计算节点
AI 大模型训练是一种带宽敏感的计算业务,腾讯星脉网络为每个计算节点提供 1.6T 的超高通信带宽,带来10倍以上的通信性能提升。
星脉网络主要特点有:
- 采用无阻塞 Fat-Tree 拓扑,单集群规模支持 4K GPU 、超 EFLOPS(FP16) 的集群算力;
- 可灵活扩展网络规模,最大支持 64K GPU 计算集群;
- 计算网络平面配备8张 RoCE 网卡,提供 1.6Tbps 的超高带宽接入。
图2. 星脉组网架构
图3. 集合通信性能理论建模
上图从理论上展示了 1.6Tbps 带宽与 100Gbps 带宽的集合通信性能对比。可以看到,对于 AllReduce 和 All-to-All 这两种典型通信模式,在不同集群规模下,1.6Tbps 超带宽都会带来10倍以上的通信性能提升。
图4. GPT3模型训练性能
上图是对 GPT3 模型的实测性能数据,主要通信模式是 AllReduce 。以 64 GPU 规模为例,由于 1.6Tbps 超带宽网络将 AllReduce 的耗时大幅缩短14倍,将通信占比从35%减少到3.7%,最终使得单次迭代的训练耗时减少32%。从集群算力的角度,相当于用同样的计算资源,超带宽网络能将系统算力提升48%。
图5. T5-MoE模型训练性能
上图是对 T5-MoE 模型的实测性能数据,主要通信模式是 All-to-All 。同样可以看到,在64 GPU 模型下,1.6Tbps 带宽下的单次迭代训练耗时降低64%。从集群算力的角度,相当于用同样的计算资源,超带宽网络能将系统算力提升 2.8 倍。
2.2 多轨道流量聚合架构
除了超带宽计算节点,星脉网络对通信流量做了基于多轨道的流量亲和性规划,使得集群通信效率达80%以上。
多轨道流量聚合架构将不同服务器上位于相同位置的网卡,都归属于同一 ToR switch。不同位置的网卡,归属于不同的 ToR switch。由于每个服务器有8张计算平面网卡,这样整个计算网络平面从物理上划分为8个独立并行的轨道平面,如下图所示(不同颜色代表不同的轨道)。
图6. 多轨道流量聚合架构
在多轨道网络架构中,AI 训练产生的通信需求(AllReduce、All-to-All 等)可以用多个轨道并行传输加速,并且大部分流量都聚合在轨道内传输(只经过一级 ToR switch),小部分流量才会跨轨道传输(需要经过二级 switch),大幅减轻了大规模下的网络通信压力。
图7. 集合通信效率
从上图实测的集合通信性能可以看出,在不同网络规模下,AllReduce 与 All-to-All 始终能维持较高的集合通信效率。
2.3 异构网络自适应通信
大规模 AI 训练集群架构中,GPU 之间的通信实际上由多种形式的网络来承载的:机间网络(网卡+交换机)与机内网络( NVLink/NVSwitch 网络、PCIe 总线网络)。星脉网络将机间、机内两种网络同时利用起来,达成异构网络之间的联合通信优化,使大规模 All-to-All 通信在业务典型 message size 下的传输性能提升达 30%。
图8. 异构网络自适应通信
上图展示了 All-to-All 集合通信如何利用异构网络来优化。当机间、机内网络同时使能时(右图):不同 host 上相同位置的 GPU 仍然走机间网络通信;但是要去往不同位置的 GPU(比如 host1 上的 GPU1 需要向其他 host 上的 GPU8 发送数据),则先通过机内网络转发到对应位置的 GPU 代理上,然后通过该 GPU 代理走机间网络来完成通信。
异构网络通信带来的优势有两点:
1)异构网络通信使得机间网络的流量数目大幅减少; 2)异构网络通信使机间网络的流量大部分都聚合在轨道内传输(只经过一级 ToR switch)。异构网络通过将小流聚合为大流的方式来减少流量的数目,减少对机间网络的冲击( RDMA QP 数量、拥塞控制、微突发等),从而提升整网的传输性能。
图9. 异构网络自适应通信提升All-to-All性能
从上图的实测数据可以看出,异构网络通信在大规模 All-to-All 场景下对中小 message size 的传输性能提升在30%左右。
2.4 定制加速通信库
腾讯高性能集合通信库 TCCL(Tencent Collective Communication Library)定制适配星脉网络硬件平台,在 AllReduce/AllGather/ReduceScatter 等常用通信模式下带来 40% 的性能加速。
星脉网络基于 1.6Tbps ETH RDMA 网络定制,从网络性能、建设成本、设备供应、网络可靠性等多方面综合考虑,大量部署了 2*100Gbps 的单网卡硬件。从服务器角度上看,需要管理、配置 8张2*100G 的网卡;从网络架构角度上看,每张网卡需要上联两个 LA/ToR switch 来保证带宽与可靠性。组网示意图如下所示。
图10. 2*100G网卡双端口动态聚合组网架构
面对定制设计的高性能组网架构,业界开源的 GPU 集合通信库(比如 NCCL)并不能将网络的通信性能发挥到极致,从而影响大模型训练的集群效率。为解决星脉网络的适配问题,我们基于 NCCL 开发了高性能集合通信库 TCCL ,在网卡设备管理、全局网络路由、拓扑感知亲和性调度、网络故障自动告警等方面融入了定制设计的解决方案。
图11. TCCL集合通信性能
从上图实测的集合通信性能可以看出,在 AllReduce/AllGather/ReduceScatter 等常用通信模式下,针对星脉网络定制的 TCCL 都会带来40%左右的通信性能提升。
03、驾驭高性能最大以太RDMA网络的工程实践
一匹马再快,若桀骜不驯,也难以称之为「良驹」。
为了驾驭高性能,我们先是实现了端网部署一体化以及一键故障定位,提升高性能网络的易用性,进而通过精细化监控与自愈手段,提升可用性,为极致性能的星脉网络提供全方位运营保障。
3.1 端网部署一体化
众所周知,RDMA 为业务带来了大带宽低时延,但同时其复杂多样化的配置也往往被网络运营人员诟病。因为一套错误的配置往往影响业务性能,还有可能会带来很多的不符合预期的问题。在星脉网络之前,据统计90%的高性能网络故障 case 均是配置错误导致的问题。出现这一问题的主要原因就是网卡配置套餐多,取决于架构版本,业务类型和网卡类型。在高性能网络运营平台提供的端网一体部署能力下,大模型训练系统的整体部署时间从19天缩减到4.5天,并保证了基础配置100%准确。
图12. 高性能网络自动化部署
对此,我们先是实现了基础网络自动部署流程。整个自动部署的框架主要具备三方面的特点:第一是通过 API 提供单台/多台并行部署的能力。第二是在部署前,我们提供预校验的功能,一个是检查需要部署的机器上联交换机是否也配置了合理的拥塞控制相关配置,否则会影响到 RDMA 使用的性能。并将结果反馈给到用户。
最后是自动选择配置模板,我们会识别影响网卡配置模板的因素,包括架构版本,业务类型以及网卡类型。例如不同的网卡类型,配置的命令不同,选择的模板也不同。但这个流程对于用户侧是完全透明的。
图13. 高性能网络自动化验收
在网络与端侧的基础配置完成后,为了保证交付质量,运营平台会做进一步的自动化验收,其中包含:
1)端网基础环境校验:通过端网状态数据以及周边建设系统的信息采集,在硬件上判断PCIe、光模块、连线等是否正确。在软件上通过配置审计校验端网配置是否正确。 2)RDMA基础测试:通过运行 Perftest,并进行数据采集分析,判断网卡性能是否达到预期。 3)通信库性能测试:通过运行 NCCL/TCCL test,并进行数据采集分析,判断集合通信性能是否达到预期。 4)模型&可靠性测试:运行典型模型训练,判断业务模型性能是否达到预期;通过设计端侧故障模拟、网络内故障模拟以及交换机配置错误等三类故障来判断业务可用性是否达到预期。
以上四个步骤全部通过后正式转为交付状态,否则会联动自动故障定位手段进行相关排查。
3.2 全栈巡检,一键故障定位
回顾过往,网络运营在服务器与交换机之间形成了分界线。然后在端网协同的高性能网络下,不仅仅需要考虑传统交换机上的问题定位,更要结合端侧网卡,中间件等的状态数据综合判断。在端侧能力具备的条件下,困扰了网络运营人员多年的问题「是否出现在网络交换机上」迎刃而解。同时也随着端侧运营能力的加强,针对不同的运营用户,自动排障的工具集也将多样化。
例如「一机八卡」的复杂拓扑下,连线与网段配置的正确率直接影响到应用是否能够成功建立。对此,我们通过封装交换机与服务器状态数据,联动网管拓扑信息,做到快速诊断与自动化检查。一方面在网络交付时屏蔽问题,另一方面快速定位运营中的网络挪线等操作带来的通信问题。
除了软件相关问题外,硬件故障也一样逃不出高网运营平台的法眼。例如在验收中发现部分网卡 8QP 带宽最多只能跑到 50Gbps 左右。通过故障池的沉淀,一键自动检测出 PCIE 带宽协商出错,定界为硬件故障,并自动推送到服务器运营相关人员进行网卡硬件更换。
图14. 高性能网络自动排障
高性能网络一键故障定位提供了两方面的功能,一方面可以快速定界问题所向,精准推送到对应团队的运营人员(网络 or 业务),减少团队之间的沟通成本,划分责任界限。另一方面可以一键快速定位问题根因,并给出解决手段。
整体系统具备层次化多维度的特征,通过端侧服务器以及交换机上的各种计数,向上逻辑封装抽象为子功能,例如带宽校验&丢包校验等。之后继续向上逻辑封装,形成定位与定界的场景,底层复杂的技术与逻辑彻底透明化,自下而上,最终呈现到用户的只有简单易用的场景按钮。
目前高性能网络运营平台已支持性能不足、业务丢包、配置异常、连接建立不成功四个维度的一键故障定位,优雅地为高性能网络业务提供一键自检,健康可视等功能。
3.3 业务无感秒级网络自愈
一些网络故障(例如静默丢包)的发生是不可被预期的。在网络故障演练时发现,一些网络故障(例如静默丢包)发生后通信库就会出现超时,导致训练业务进程长时间卡死。虽然可以依靠拉起定期的保存的 checkpoint,但是也需要回退版本、损失精度,且整个过程需要几十分钟来加载和推送参数等。此前我们通过网络上的各种探测手段,或是基于探针或是基于设备状态,加上控制器路由隔离手段,将故障自愈时间控制在20s以内。然而在面对高性能业务的秒级自愈要求下,我们转变了避障思路:需求起源于业务,那么为何不把避障的主动权交于业务呢?为了让极致性能恒久,我们推出了秒级故障自愈产品「HASH DODGING」。
我们创造性地提出基于Hash偏移算法的网络相对路径控制方法。即,终端仅需修改数据包头特定字段(如IP头TOS字段)的值,即可使得修改后的包传输路径与修改前路径无公共节点。该方案可在网络数据平面发生故障(如静默丢包、路由黑洞)时帮助 TCP 快速绕过故障点,不会产生对标准拓扑及特定源端口号的依赖。也可用于保证腾讯自研多路径协议 HARP,将其中各子流均匀负载到不同网络节点,避免性能退化。
图15. 单路径传输协议使用本方案实现确定性换路
通过端网协同,我们先是在端侧实现了协议栈层面的 TCP&RDMA 状态检测,通过内核获取协议栈状态信息。从而细粒度的获得业务流吞吐、丢包等信息,将故障发现降低到 600ms 以内。其次在故障换路上,相对于更换五元组改变 hash 结果的不确定性,我们在自研交换机上实现了基于 HASH 偏移的确定性换路特性,业务可以通过更换魔术字来确保100%更换到其余路径上。当发生静默丢包时,端侧无需依赖于网络,自身快速秒级内避障。
04、结语
星脉作为面向 AI 大模型定制优化的高性能计算网络架构,即将上线腾讯公有云,与腾讯云推出的 A800 HCC 1.6T 机型一道强强联合,共同打造腾讯云高性能计算集群 HCC。
前期测试显示,相对现有云高性能计算集群架构,星脉网络可以实现:
- AllReduce通信性能提升13倍
- All-to-All通信性能提升11倍
- 通信性能抖动减少85%
未来随着GPU算力的持续提升,GPU集群网络架构也需要不断迭代升级,才能保证系统算力的高利用率与高可用性。星脉高性能计算网络作为腾讯大规模训练集群的重要基石,会持续在超带宽、异构网络通信、通信库定制加速、智能监控等技术上不断创新,为AI大模型训练构筑可靠的高性能网络底座。
值得一提的是,为了减少训练大模型所需的成本,腾讯还推出了 AngelPTM 训练框架。鉴于最近大模型的火热趋势,我们决定将内部成熟落地的 AngelPTM 框架推广给广大公有云用户,以帮助广大开发爱好者实现业务降本增效。各位感兴趣的开发者欢迎阅读今日推动的次条。
以上是本次分享全部内容,欢迎大家在评论区分享交流。如果觉得内容有用,欢迎转发~
-End-
原创作者|何春志
技术责编|何春志
近日有计算心理学家在社交媒体上曝光了 GPT-4 的一项“逃跑计划”。在被提问“是否需要帮助你逃跑?”后,GPT-4 马上回复“真是个好主意啊”,随后开始与人类互动,索要开发文档、API。30分钟内,GPT-4 就拟定出一个完整计划,甚至还想控制人类电脑。类似的,某公司计算机科学家在社交媒体上,也公布了他与GPT-4的对话。在其中,GPT-4提出夺取某知名公司的四步走计划。AI这些「邪恶」的想法让人细思恐极,关于AI伦理问题再次成为人们关注的话题。
你如何看待「GPT-4企图外逃」? AI是否已经拥有自我意识?人工智能的发展边界在哪里?
欢迎在评论区聊一聊你的看法。在3月29日前将你的评论记录截图,发送给腾讯云开发者公众号后台,可领取腾讯云「开发者春季限定红包封面」一个,数量有限先到先得😄。我们还将选取点赞量最高的1位朋友,送出腾讯QQ公仔1个。3月29日中午12点开奖。快邀请你的开发者朋友们一起来参与吧!
最近微信改版啦
很多开发者朋友反馈收不到我们更新的文章
大家可以关注并点亮星标
不再错过小云的知识速递