arXiv · 2309.13541
Efficient All-to-All Collective Communication Schedules for Direct-Connect Topologies
Abstract
The all-to-all collective communications primitive is widely used in machine learning (ML) and high performance computing (HPC) workloads, and optimizing its performance is of interest to both ML and HPC communities. All-to-all is a particularly challenging workload that can severely strain the underlying interconnect bandwidth at scale. This paper takes a holistic approach to optimize the performance of all-to-all collective communications on supercomputer-scale direct-connect interconnects. We address several algorithmic and practical challenges in developing efficient and bandwidth-optimal all-to-all schedules for any topology and lowering the schedules to various runtimes and interconnect technologies. We also propose a novel topology that delivers near-optimal all-to-all performance.
Explore related subjects
Keep this discovery
Prithwish Basu, Liangyu Zhao, Jason Fantl, Siddharth Pal, Arvind Krishnamurthy, Joud Khoury. 2023-09-24. Efficient All-to-All Collective Communication Schedules for Direct-Connect Topologies. https://arxiv.org/abs/2309.13541
Cite the original work for its findings. Save a collection to share your selection of sources.