arXiv:2512.10236cs.DCcs.AR2025-12被引 2

提出细粒度计算通信重叠方法FiCCO,提升多GPU训练效率。

Design Space Exploration of DMA based Finer-Grain Compute Communication Overlap

  • 在张量分片基础上进一步细分,实现更细粒度的计算通信重叠
  • 通过启发式调度在真实场景中最高提速1.6倍,81%未见场景选对最优方案
  • 将通信卸载至GPU DMA引擎,减少操作重叠导致的竞争瓶颈

现代机器学习工作负载需要在多个GPU间分布训练与推理。然而,现有并行化技术常暴露关键路径通信问题,导致计算-通信重叠潜力未被充分利用,最多损失1.7倍性能。已有重叠方法基于模型状态和输入的分片结构,在分片粒度上实现重叠,但受限于网络拓扑支持与数据流效率。本文提出细粒度重叠方法FiCCO,其在传统分片基础上进一步细化,拓展了执行调度的设计空间,支持更多网络拓扑并实现更精细的数据流。我们系统分析重叠中的性能瓶颈,发现分解与竞争是主要限制因素,并将其与静态算子大小相关联,据此设计可被框架和运行时采用的启发式策略,根据操作特性选择定制化调度。为缓解操作重叠带来的竞争,我们将通信卸载至GPU DMA引擎。在多种真实部署场景中评估,所提启发式策略驱动的定制调度最高可实现1.6倍加速,且在81%未见过的场景中准确选出最优调度。

原文摘要 · Abstract (English)

Modern ML workloads demand distributing training and inference across multiple GPUs. However, these parallelization techniques often suffer from exposed critical-path communication, leaving a potential 1.7x speedup on the table through compute-communication overlap. Prior overlapping methods harness the fact that ML model state and inputs are already sharded into the number of GPUs, and overlap the compute and communication at shard granularity. However, such coarse-grained overlap suffers from limited network topology support, and suboptimal dataflows. In this work, we instead make a case for finer-grain compute-communication overlap which we term FiCCO. FiCCO operates one level deeper than traditional sharding, and unlocks overlap for a wider set of network topologies and enables finer-grain dataflow. We show that FiCCO opens up a wider design space of execution schedules than possible at shard-level alone. To walk the design space of schedules, we study and characterize the performance inefficiencies on doing overlap and overlay the schedules with the associated inefficiency signatures. Our characterization reveals decomposition and contention based slowdowns to be the major performance limiters, and we correlate the slowdown factors with the static compute/communication operator sizes. This helps us design heuristics (that frameworks and runtimes can harness) to select bespoke FiCCO schedules based on the nature of underlying ML operations. Finally, to further minimize contention inefficiencies inherent with operation overlap, we offload communication to GPU DMA engines. We evaluate several scenarios from realistic ML deployments and demonstrate that our proposed heuristics driven bespoke schedules deliver up to 1.6x speedup. Further, our heuristics provide accurate guidance to pick the optimal schedule in 81% of unseen scenarios.

多GPU通信重叠调度优化DMA

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