LLaMCAT: Optimizing Large Language Model Inference with Cache Arbitration and Throttling
Large Language Models (LLMs) have achieved unprecedented success across various applications, but their substantial memory requirements pose significant challenges to current memory system designs, especially during inference. Our work targets last-level cache (LLC) based architectures, including GPUs (e.g., NVIDIA GPUs) and AI accelerators. We introduce LLaMCAT, a novel approach to optimize the LLC for LLM inference. LLaMCAT combines Miss Status Holding Register (MSHR)- and load balance-aware cache arbitration with thread throttling to address stringent bandwidth demands and minimize cache stalls in KV Cache access. We also propose a hybrid simulation framework integrating analytical models with cycle-level simulators via memory traces, balancing architecture detail and efficiency.
Authors
- Zhongchun Zhou
- Wei Zhang
- Chengtao Lai
Institutions
- Hong Kong University of Science and Technology (HK)
Publication Details
- Published
- 2025-09-08
- DOI
- https://doi.org/10.1145/3754598.3754671
- Primary Topic
- Parallel Computing and Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00