arXiv:2605.31000cs.NIcs.LG2026-05

解决多厂商异构集群通信瓶颈,提升大模型训练效率

HetCCL: Enabling Collective Communication For Mixed-Vendor Heterogeneous Clusters

论文配图:HetCCL: Enabling Collective Communication For Mixed-Vendor Heterogeneous Clusters
图 1 · 摘自论文原文
  • 通过设备间直接通信避免主机内存拷贝,降低通信开销
  • 在异构环境下实现17-19倍于Gloo的通信带宽,端到端训练提速16.9%
  • 支持多厂商硬件,适用于大规模分布式AI训练场景

在异构集群上训练大语言模型(LLMs)面临集体通信的重大挑战,因为多厂商硬件引入了不同的网络和计算特性。现有集体通信框架(如NCCL、RCCL)针对同质环境设计,无法适应混合硬件配置;而具备异构支持的通信库(如Gloo、OpenMPI)在数据路径中产生显著开销。本文提出HetCCL,一种通过异构设备间的高效点对点传输实现异构集体通信的框架,避免主机与设备间的内存拷贝,同时将控制逻辑卸载至CPU。对于组合型集体操作(如AllReduce、ReduceScatter),HetCCL引入边界通信器机制,利用各厂商通信库内建的归约能力实现厂商无关性。结合高效的异构点对点传输与可移植的归约机制,HetCCL提出分层拓扑抽象,将集体通信分解为集群级原语,确保跨集群数据传输量最优且带宽利用率最大化。我们实现了支持4种厂商的HetCCL,在4种异构设置下进行了基准测试和端到端大模型任务评估。结果表明,相较于Gloo,HetCCL在异构通信中实现了17-19倍的带宽提升,端到端训练每步时间最快缩短16.9%。

原文摘要 · Abstract (English)

Training Large Language Models (LLMs) on heterogeneous clusters presents significant challenges for collective communication, as hardware from multiple vendors introduces diverse network and computational characteristics. Existing collective communication frameworks (e.g., NCCL, RCCL) designed for homogeneous environments fail to address mixed-hardware setups, while communication libraries with heterogeneous support (e.g., Gloo, OpenMPI) incur heavy overhead in the data path. This paper presents HetCCL, a framework that enables heterogeneous collective communication by efficient P2P transport across heterogeneous devices (e.g., GPUs), eliminating the host-device memory copy overhead while offloading the control to the CPUs. For combining collectives (e.g., AllReduce, ReduceScatter), HetCCL introduces a border-communicator mechanism that achieves vendor independence by using the intrinsic reduction in the combining collectives in vendor collective communication libraries. With efficient heterogeneous P2P transport and portable reduction mechanism, HetCCL proposes a hierarchical topology abstraction for heterogeneous clusters, dissecting collective communication into cluster-level primitives that guarantee optimal cross-cluster data transfer volume and optimal bandwidth utilization. We implement HetCCL with 4 different vendor support and evaluate it in 4 heterogeneous settings with benchmarks and end-to-end LLM tasks. Our evaluation shows that HetCCL achieves 17-19x higher bandwidth than Gloo in heterogeneous communications, and speeds up end-to-end training by up to 16.9% in the per-step-time.

异构通信大模型训练集体通信GPU集群

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。