Search papers, labs, and topics across Lattice.
This paper introduces MultiWrite, a novel many-to-many transmission semantic for collective communication that leverages multicast principles to eliminate redundant packets in AllGather and AlltoAll operations. MultiWrite addresses the limitations of traditional multicast, such as management overhead and compatibility issues, for AI workloads. Implemented on Ascend NPUs, MultiWrite achieves up to 33% latency reduction in long-term stress tests on commercially deployed devices.
Multicast, re-imagined for AI, slashes collective communication latency by up to 33% on real-world hardware.
Reducing collective communication latency is a critical goal for large model training and inference in both academia and industry. Many-to-many communications, such as AllGather and AlltoAll (dispatch), are core components of modern parallelization strategies. State-of-the-art implementations of these communications rely on unicast-based writes and transmit duplicate copies of the same data across physical links for multiple receivers. This redundant transmission congests network bottlenecks and degrades end-to-end latency. We present MultiWrite, a novel many-to-many transmission semantic that eliminates redundant packets to directly reduce operator latency. MultiWrite adopts multicast principles while addressing critical limitations of traditional multicast for AI workloads. These limitations include heavy management plane overhead and ecosystem compatibility issues. We implement MultiWrite on Ascend NPUs. Long-term stress tests demonstrate that our MultiWrite-based operators achieve up to 33% latency reduction on commercially deployed devices.
