AI & ML
FastAFD: When Does Attention-FFN Disaggregation Pay on Rack-Scale NVLink?
arXiv cs.DC0 views
arXiv:2611.07149v1 Announce Type: new Abstract: Attention-FFN disaggregation (AFD) separates attention from the experts of a mixture-of-experts (MoE) model and has been reported to raise decode throughput over colocated serving. Whether this advantage holds on rack-scale NVLink systems such as GB200 NVL72 remains unclear: prior results were obtained on other hardware, against narrower colocated baselines, or with analytical models, whereas production engines serve large MoE models on these systems with wide expert parallelism (wide EP). We build and tune FastAFD, an AFD engine for GB200 NVL72 that overlaps the MoE of one microbatch with the attention of the other, and compare it against the best vLLM wide EP setting. Experiments show that AFD has a throughput advantage of up to 1.17x over wide EP on Qwen3 and 1.16x on Qwen3.5, and matches it on MiniMax-M2.5. We derive an analytical model that predicts AFD throughput within 3% using only profiles of colocated vLLM, telling whether switching to AFD would pay off before deploying it. The model yields a concise insight: AFD's speedup factors into a batch gain (larger batches, since attention GPUs hold no expert weights), an overlap gain (shorter steps, since the MoE runs on the expert GPUs behind attention), and a disaggregation cost (the GPUs given to experts). AFD has the advantage when the two gains outweigh the cost. This view explains which models benefit from AFD, and shows that speculative decoding helps light-expert models most, where the enlarged overlap gain outweighs the larger disaggregation cost and adds a further 8% to AFD's advantage over wide EP. Looking ahead, serving the experts on accelerators that keep their weights in on-chip SRAM, such as the Groq LPX accelerators paired with Vera Rubin NVL72, would remove most of the disaggregation cost, raising the ideal speedup to 2.48x with speculative decoding. FastAFD is open source at https://github.com/hao-ai-lab/FastAFD.
Source:https://arxiv.org/abs/2611.07149PreviousNUMA-Torch: Compiler-Driven NUMA-Aware LLM Prefill on Multi-Socket CPUsNextCapability Gain from ISA to Shared Execution on AMD MI300A and MI250X
Related posts
Sharp Tech on Video: The U.S. Votes on Data Centers, Implications for Big Tech, Google’s Response to AI
0Stratechery

I ran Switch headless on a cloud box so my team could pick one pull request out of four
0dev.to GitHub

Why the Meeting Room Whiteboard Still Says the Wi-Fi Password Every Week
0dev.to Discuss
I stopped prompting harder. I started engineering the loop.
0dev.to Productivity
How to Calculate Your Real Hourly Rate as a Freelancer
0dev.to Productivity

What osFoundry Maestro does, an AI chat that shows every step it took
0dev.to Productivity