English

FusionCIM: Accelerating LLM Inference with Fusion-Driven Computing-in-Memory Architecture

Hardware Architecture 2026-04-29 v1

Abstract

In this paper, we propose FusionCIM, an operator-fusion-driven compute-in-memory (CIM) accelerator architecture for efficient and scalable LLM inference, with three key innovations: (1) a hybrid CIM pipeline architecture that maps QKT computation on inner-product-based CIM (IP-CIM) and PV aggregation on outer-product-based CIM (OP-CIM) for efficient matrix multiplications fusion; (2) a QO-stationary dataflow that eliminates repeated KV loading in CIM and K-matrix access in buffer under transpose fusion, significantly improving data reuse on chip; and (3) a pattern-aware online-softmax mechanism that exploits distribution regularities of attention scores to reduce exponential rescaling overhead for non-linear fusion. Experimental results on LLaMA-3 model show that FusionCIM achieves up to 3.86x energy saving, and 1.98x speedup compared with prior SOTA CIM-based designs with 29.4 TOPS/W energy efficiency at the system level.

Keywords

Cite

@article{arxiv.2604.25317,
  title  = {FusionCIM: Accelerating LLM Inference with Fusion-Driven Computing-in-Memory Architecture},
  author = {Zihao Xuan and Jia Chen and Yewen Li and Wei Xuan and Hegan Chen and Xiao Huo and Fengbin Tu},
  journal= {arXiv preprint arXiv:2604.25317},
  year   = {2026}
}

Comments

7 Pages, 10 figures

R2 v1 2026-07-01T12:38:40.373Z