Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper revisits Bruck's All-to-All communication algorithm in the context of optical reconfigurable networks (ORNs) to improve performance for distributed ML and HPC workloads. They introduce ReTri, a bidirectional All-to-All schedule based on balanced ternary block propagation that completes communication in $\lceil \log_3 n\rceil$ phases. Simulations demonstrate that ReTri achieves up to 10x speedup over static All-to-All and 2.1x over reconfigurable Bruck, even with millisecond-scale reconfiguration delays.
All-to-all communication can be sped up by 10x in distributed ML/HPC by co-designing communication patterns and optical network reconfiguration strategies.
All-to-All communication is a key performance bottleneck for distributed machine learning (ML) and high-performance computing (HPC) workloads, where dense traffic increasingly stresses scale-up interconnects. While these ML and HPC workloads have driven unprecedented infrastructure demand, optical reconfigurable networks (ORNs) offer a promising path forward. By adapting the physical topology to the active workload, they improve communication cost and bandwidth utilization. However, their benefit is critically contingent on whether the collective consists of structured phases that can be served by sparse and reusable topology states. In this paper, we revisit Bruck's All-to-All implementation and demonstrate the benefits of topology optimization in which both communication pattern and reconfiguration strategy are co-designed. We present ReTri, a bidirectional All-to-All schedule for ORNs. ReTri uses balanced ternary block propagation to complete All-to-All in $\lceil \log_3 n\rceil$ phases. The induced reconfiguration strategy from ReTri's pairwise bidirectional exchanges allow reconfiguration delays to be amortized across multiple phases. Preliminary simulations show that ReTri improves completion time by up to $10\times$ over static All-to-All, even for millisecond-scale reconfiguration delays, and improving reconfigurable Bruck by up to $2.1\times$.