Eliminating Hidden Serialization in Multi-Node Megakernel Communication
On the map: Mixture-of-experts → Efficient MoE training and inference systems
Abstract (arXiv)
Recent megakernel designs for Mixture-of-Experts (MoE) inference fuse expert computation with fine-grained, GPU-initiated communication into a single persistent GPU kernel, and outperform collective-based MoE on a single node by overlapping data transfer with compute at tile granularity. This benefit does not carry over cleanly to multi-node inference, where experts span many nodes connected by an RDMA fabric. Communication-bound MoE models regress by up to 10× on 8 nodes, and the regression worsens with node count. We trace this regression to hidden serialization in proxy-based RDMA transports. The ordering requirement between each tile transfer and its completion signal forces a fence that drains the NIC pipeline, and its cost grows with the number of concurrent transfers. As a result, models whose per-expert compute is too small to absorb this inflated network latency expose communication on the critical path. We present Perseus, which eliminates this serialization through two techniques. Decoupled signaling batches fences at per-destination granularity, reducing fence count by 8×. NIC-side ordering replaces proxy stalls with hardware fence flags, so the proxy never blocks. On proxy-based transports, Perseus achieves up to 10.3× end-to-end speedup. Perseus on IBRC matches or exceeds IBGDA GPU-direct by up to 1.2×, which shows that serialization, rather than the choice between proxy-based and GPU-direct transport, is what bounds multi-node megakernel performance.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Efficient MoE training and inference systems, one of 12 directions in Mixture-of-experts. First posted on arXiv 1 May 2026.
Builds on
No reference list yet.
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Fine-grained Computation-Communication Overlap via Tile-level Signaling and Scheduling for Mixture-of-Experts
- 2026Expert Coupling in MoE Pretraining: Reducing All-to-All Overhead with Correlated Placement and Token Shuffling
- 2026Weave: Fine-Grained Dynamic SM Scheduling in an MoE Megakernel for Compute-Communication Overlap
- 2026Efficient Mixture-of-Experts with Speculative Decoding via Expert Coactivation
- 2025Occult: Optimizing Collaborative Communication across Experts for Accelerated Parallel MoE Training and Inference
- 2026Federation of Experts: Communication Efficient Distributed Inference for Large Language Models
- 2026Democratizing MoE inference on commodity GPUs with CoMoE
- 2026Relay Buffer Independent Communication over Pooled HBM for Efficient MoE Inference on Ascend