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
Looking for first-time contributors: Open Catalog of AI System Prompts
0dev.to GitHub
How do you reduce LLM spam in PR reviews?
0dev.to GitHub

GH Profile Cyber Punk City Spin
0dev.to GitHub
Lost Your GitHub Repository? How to Privately Investigate and Safeguard Development Quality
0dev.to GitHub
FCC router ban: Netgear, Eero and Asus win Conditional Approval, TP-Link has not
0dev.to News
GPT Image 2.5 or Nano Banana 2.1? Cost per usable image
0dev.to Design