arXiv:2504.09014cs.DCcs.AI2025-04被引 16

MSCCL++让AI推理通信更快更通用,支持新硬件只需数周。

MSCCL++: Rethinking GPU Communication Abstractions for AI Inference

  • 提供简洁接口暴露关键硬件特性,隐藏同步复杂性
  • 集体通信速度提升1.7倍(最高5.4倍),推理性能提升1.2倍(最高1.38倍)
  • 适合追求高性能且不愿手写通信代码的开发者

AI应用越来越多地部署在快速演进的异构硬件上以实现最佳性能,但通用库对这些特性的支持滞后。性能导向的程序员常自行构建通信栈,虽快但易出错且难以移植。本文提出MSCCL++,一种高性能、可移植通信内核的设计方法:(1) 提供低层级、保留性能的原语接口,仅暴露最小硬件抽象并隐藏同步与一致性复杂性;(2) 提供高层领域特定语言(DSL),供开发者实现定制化通信算法;(3) 内置高效算法库,实现标准集体通信接口,使无经验用户也能快速采用。相比最先进基线,MSCCL++在集体通信上实现几何均值1.7倍(最高5.4倍)加速,在AI推理工作负载上实现1.2倍(最高1.38倍)加速。MSCCL++已用于微软Azure多个AI服务,并被AMD维护的RCCL通信库采纳。项目开源,地址为https://github.com/microsoft/mscclpp。两年实践经验表明,其抽象设计稳健,支持如multimem等新硬件功能的开发可在数周内完成。

原文摘要 · Abstract (English)

AI applications increasingly run on fast-evolving, heterogeneous hardware to maximize performance, but general-purpose libraries lag in supporting these features. Performance-minded programmers often build custom communication stacks that are fast but error-prone and non-portable. This paper introduces MSCCL++, a design methodology for developing high-performance, portable communication kernels. It provides (1) a low-level, performance-preserving primitive interface that exposes minimal hardware abstractions while hiding the complexities of synchronization and consistency, (2) a higher-level DSL for application developers to implement workload-specific communication algorithms, and (3) a library of efficient algorithms implementing the standard collective API, enabling adoption by users with minimal expertise. Compared to state-of-the-art baselines, MSCCL++ achieves geomean speedups of $1.7\times$ (up to $5.4\times$) for collective communication and $1.2\times$ (up to $1.38\times$) for AI inference workloads. MSCCL++ is in production of multiple AI services provided by Microsoft Azure, and has also been adopted by RCCL, the GPU collective communication library maintained by AMD. MSCCL++ is open source and available at https://github.com/microsoft/mscclpp . Our two years of experience with MSCCL++ suggests that its abstractions are robust, enabling support for new hardware features, such as multimem, within weeks of development.

通信优化GPUAI推理开源

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