2. GPU集群互连需求分析

互连网络的设计不能从“选什么交换机”开始,而必须从“任务需要什么”开始。同一张GPU集群网络,跑大模型训练、跑CFD/HPC、跑在线推理,对带宽、延迟、扩展性和成本的压力点完全不同。本节的任务是把这三类负载翻译成可量化的网络需求,形成后续拓扑选型、拥塞控制和成本权衡的输入。

一、三类负载的通信特征

AI训练以数据并行、张量并行、流水线并行和专家并行混合为主。数据并行在每次迭代末尾做梯度AllReduce,通信量与参数量成正比;张量并行在层内频繁做AllReduce或AllGather,对延迟和带宽都敏感;流水线并行产生周期性点对点激活传递,容易形成突发;专家并行(MoE)则产生All-to-All,是典型的“多对多”模式,对网络对分带宽和负载均衡要求最高。训练任务通常持续数小时到数周,追求的是高持续带宽、低尾延迟、可预测的完成时间。

HPC以MPI通信为核心,典型模式包括邻居交换(stencil)、全局归约(AllReduce)、快速傅里叶变换中的All-to-All。HPC对低延迟极为敏感,因为很多通信是细粒度、小消息、高频次的;同时HPC常要求可扩展性接近线性,即节点数翻倍时性能仍能按比例提升。HPC还常涉及CPU-GPU协同,网络需要同时承载主机侧和设备侧流量。

推理分为离线批推理和在线服务推理。离线批推理通信模式接近训练但规模更小;在线推理以请求-响应为主,单次通信量小,但对尾延迟(P99)极其敏感,因为用户可感知。推理集群还常采用多副本、负载均衡和KV Cache传输,网络需要支持高并发小消息和突发流量。

三类负载的通信模式与网络需求映射 AI 训练 HPC 推理 数据并行 AllReduce 梯度同步,周期性强 张量并行 AllGather 层内高频,延迟敏感 流水线 P2P 激活 突发性强,易拥塞 专家并行 All-to-All 多对多,对分带宽关键 核心需求 持续高带宽(≥400G/GPU) 低尾延迟,抖动小 可预测完成时间 支持大规模扩展 成本:带宽/美元优先 邻居交换 Stencil 小消息,高频次 全局归约 AllReduce 同步点,延迟敏感 FFT All-to-All 全局重排,带宽密集 CPU-GPU 协同 主机+设备混合流量 核心需求 超低延迟(<2μs) 线性可扩展性 小消息高吞吐 确定性通信 成本:延迟/美元优先 请求-响应 小消息,高并发 KV Cache 传输 大块数据,突发 多副本负载均衡 流量分散,动态 批处理聚合 吞吐与延迟权衡 核心需求 P99 尾延迟优先 高并发连接数 弹性扩缩容 成本:请求/美元优先 能效比重要

二、带宽需求的量化方法

带宽需求可以从“单GPU出口带宽”和“网络对分带宽”两个层面估算。以训练为例,假设模型参数量为P,采用数据并行,梯度用FP16存储,则一次AllReduce需要传输约2P字节(Reduce-Scatter + AllGather)。若每步计算时间为Tcomp,则梯度同步所需带宽为2P/Tcomp。例如P=70B(700亿参数),FP16下梯度约140GB,若每步计算耗时200ms,则AllReduce需要约140GB/0.2s=700GB/s的聚合带宽。若集群有512块GPU,每块GPU平均需要约1.37GB/s的出口带宽用于梯度同步——这还没算张量并行和流水线的额外流量。实际工程中,单GPU出口带宽通常按400Gbps(50GB/s)配置,留出余量应对突发和混合并行。

对分带宽(bisection bandwidth)是衡量网络整体能力的关键指标。对于N个节点的胖树,对分带宽约为N/2×单链路带宽。若集群有1024块GPU,每块400Gbps,则对分带宽至少需要1024/2×400Gbps=204.8Tbps。若采用Rail-Optimized拓扑,对分带宽会降低,但成本也下降,适合通信模式以邻居交换为主的HPC场景。

三、延迟约束与可扩展性

延迟分为链路延迟、交换延迟和协议栈延迟。GPU集群中,RDMA over Converged Ethernet(RoCE)或InfiniBand的端到端延迟通常在1.5~3μs。对HPC的细粒度通信,若单次通信延迟为2μs,而计算时间为1μs,则通信开销占比超过60%,必须通过计算通信重叠或降低延迟来优化。对训练,AllReduce的延迟主要影响同步点,若每步有多次AllReduce,累积延迟会显著拉长迭代时间。

可扩展性常用等效率或等速度模型衡量。理想情况下,节点数翻倍,性能翻倍;但网络拥塞、协议开销和同步等待会使实际加速比下降。工程上关注“扩展效率”,即N节点加速比/N。若扩展效率低于80%,就需要检查网络是否成为瓶颈。对推理,可扩展性更多体现为副本数增加时,P99延迟是否保持稳定。

四、成本约束与权衡

成本包括交换机端口成本、光模块成本、线缆成本和运维成本。以400G端口为例,交换机单端口成本约500~1000美元,光模块约300~600美元,线缆约50~200美元。若集群有1024块GPU,每块需要1~2个400G端口,则网络硬件成本约在150万~300万美元。若采用800G端口,端口密度翻倍,但光模块和交换机成本也上升,需要根据带宽需求判断是否值得。

成本优化的核心是“按需分配”。训练集群可以接受较高的对分带宽和较低收敛比(如1:1),因为AllReduce和All-to-All对带宽敏感;HPC集群可以采用收敛比1:2或1:3,因为邻居交换的局部性较强;推理集群可以采用收敛比1:4甚至更高,因为请求-响应流量分散且可容忍一定排队。此外,推理集群可以利用CPU-GPU异构和模型量化降低单请求带宽,从而降低网络成本。

量化示例(假设):某训练集群512块GPU,模型70B,FP16梯度,每步计算200ms。梯度AllReduce聚合带宽需求约700GB/s,单GPU出口约1.37GB/s。若配置400Gbps(50GB/s)出口,余量约36倍,可覆盖张量并行和流水线突发。对分带宽按1:1收敛比需512/2×400Gbps=102.4Tbps,采用两层胖树,每层64端口400G交换机,可满足需求。网络硬件成本估算约120万~200万美元。

五、需求到设计的映射

把上述需求整理成设计输入,可以形成一张“需求-指标-设计”对照表。训练任务优先保证对分带宽和低尾延迟,适合胖树或Dragonfly+;HPC优先保证低延迟和线性扩展,适合胖树或Torus;推理优先保证P99延迟和弹性,适合叶脊架构加负载均衡。成本约束则决定收敛比、端口速率和光模块类型。后续章节将围绕这些输入,展开拓扑选型、路由策略和拥塞控制的具体设计。

负载类型 关键通信模式 带宽需求 延迟需求 扩展性目标 成本优先级
AI训练 AllReduce、All-to-All、P2P 持续高带宽,单GPU≥400Gbps 低尾延迟,抖动小 千卡以上线性扩展 带宽/美元
HPC Stencil、AllReduce、FFT 小消息高吞吐,对分带宽适中 超低延迟(<2μs) 万卡级线性扩展 延迟/美元
推理 请求-响应、KV Cache 中等带宽,高并发 P99尾延迟优先 弹性扩缩容 请求/美元、能效