阅读研究雷达投资体系
登录 / 注册
← 返回阅读
资讯快讯2 分钟阅读DisconfirmAI 编辑部

英伟达:本地AI集群何时该用交换机而非环形拓扑

英伟达公布DGX Spark八节点集群基准数据,并解释了交换机拓扑与环形拓扑之间的性能分界机制:小消息、少节点时all-to-all交换减少通信轮次、降低延迟;消息或节点数增大后带宽开销上升,环形的转发效率反而更优。该判断适用于基于RoCE互联的本地推理集群设计。

2026-10-23 00:00

英伟达在2026年10月柏林GTC大会上宣布DGX Spark正式支持八节点集群,统一内存从单节点128 GB扩展到八节点1 TB,使本地桌面可以运行GLM 5.3(753B)等前沿级模型。但比内存容量更值得关注的,是这篇文章给出的拓扑选择机制——它回答了一个实际工程问题:搭建本地多节点推理集群时,什么时候需要交换机,什么时候环形直连就够了。

英伟达在八节点集群上测试了一种名为RoCEnante的集合通信后端。它的原理是all-to-all交换:每个节点将自己的输入直接发给所有其他节点,然后在本地完成归约,整个过程由一个GPU内核协调。相比环形all-reduce需要数据在相邻节点间逐跳转发多轮,all-to-all把通信轮次压缩到一步,因此在小消息、低并发、节点数较少的张量并行解码场景下能降低延迟。

但代价是数据搬运量更大。每个节点都要把完整输入发给所有对等节点,并在本地对全部输入做归约。当消息尺寸或节点数增长时,这份额外流量变得昂贵,环形转发在带宽效率上的优势重新显现。英伟达的测试数据印证了这个分界:在交换机拓扑上使用RoCEnante,八节点解码吞吐达到52.58 TPS(相对单节点4.43倍加速),而同一拓扑不用RoCEnante时仅3.05倍加速。在四节点NVFP4精度下,RoCEnante带来约21%的解码吞吐提升;在八节点非量化模型上,这一提升降至约18%,与上述延迟—带宽权衡一致。

英伟达明确标注了适用边界:RoCEnante仅适用于交换机拓扑,因为它依赖节点间的直接点对点连接;当需要高并发、更大集群或环形拓扑时,NCCL仍是更优选择。该后端已合入FlashInfer,启动测试失败时自动回退到NCCL。原始one-shot all-reduce方案由社区贡献者Jason Cook(el8)提出。

对开发者的实际判断依据是:四节点推理用环形直连即可,成本最低、设置最简单,NCCL默认的环形算法与这种拓扑天然匹配。需要扩展到八节点、将集群拆分为独立模型池(如用Switchyard做多模型路由)、或运行MPI等需要任意点对点通信的工作负载时,才需要交换机。英伟达的测试条件为并发1、8192 token输入、256 token输出、NVFP4量化精度,实际工作负载偏离这些条件时分界点可能移动。

来源

  • developer.nvidia.comNVIDIA Technical Blog↗
带入研究 →
登录 / 注册
阅读研究雷达投资体系