AI & ML
AMMA: A Multi-Chiplet Memory-Centric Architecture for Low-Latency 1M Context Attention Serving
arXiv cs.DC0 views
arXiv:2604.26103v3 Announce Type: replace-cross Abstract: All current LLM serving systems place the GPU at the center, from production-level attention-FFN disaggregation to NVIDIA's Rubin GPU-LPU heterogeneous platform. Even academic PIM/PNM proposals still treat the GPU as the central hub for cross-device communication. Yet the GPU's compute-rich architecture is fundamentally mismatched with the memory-bound nature of decode-phase attention, inflating serving latency while wasting power and die area on idle compute units. The problem is compounded as reasoning and agentic workloads push context lengths toward one million tokens, making attention latency the primary user-facing bottleneck. To address these inefficiencies, we present AMMA, a multi-chiplet, memory-centric architecture for low-latency long-context attention. AMMA replaces GPU compute dies with HBM-PNM cubes, roughly doubling the available memory bandwidth to better serve memory-bound attention workloads. To translate this bandwidth into proportional performance gains, we introduce (i) a logic-die microarchitecture that fully exploits per-cube internal bandwidth for decode attention under a minimal power and area budget, (ii) a two-level hybrid parallelism scheme, and (iii) a reordered collective flow that reduces intra-chip die-to-die communication overhead. We further conduct a design-space exploration over per-cube compute power and intra-chip D2D link bandwidth, providing actionable guidance for hardware designers. Evaluations show that AMMA achieves 15.5X lower attention latency and 6.9X lower energy consumption compared with the NVIDIA H100.
Source:https://arxiv.org/abs/2604.26103PreviousPorting and Benchmarking Chapel on Emerging RISC-V Hardware: An HPC Viability StudyNextTileMaxSim: Roofline Regimes and Compression Limits for Fused GPU MaxSim Scoring
Related posts
HN 1위 'Subvocal — 소리 없이 컴퓨터와 대화하기'가 594 업보트를 받으며 개발자 커뮤니티를 달궜습니다. 서브보컬(아성대 근육 미세 신호)만으로...
0dev.to UX
Phân tích dữ liệu Cagliari 1-0 Frosinone: Hiệu quả tối đa của Adopo và điểm tựa Caprile
0dev.to Data

Phán Quyết Kỷ Luật Tại Giải VĐQG Morocco: Biến Động Điểm Số Của Widad Temara Và Wydad Casablanca
0dev.to Data
Switching Recovery Tools Is a Data Job Before It Is a Vendor Job
0dev.to Data
⚡ What Every Developer Forgets!
0dev.to Discuss
Moonshot's Kimi K3.1 Pro Shakes Up Western AI Coding
0dev.to Machine Learning