LLM-CIM: A 28nm 126.7TOPS/W Input-LUT-Based Digital CIM Macro with Reconfigurable Matrix Multiplication and Nonlinear Operation Modes for LLMs
This paper presents a Digital Computing-in-Memory (DCIM) macro tailored for Large Language Model (LLM) acceleration, named LLM-CIM. It has three key features: 1) A Maximum Reuse-Oriented Input-LUT (MRIL) CIM reduces compute logic area and power by 40.7 % and 37.9 %. 2) An Outlier Aggregation-based Reordering Unit (OARU) saves <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$1.68 \sim 1.72\mathrm{x}$</tex> CIM computation time. 3) A Normalized-Domain Computation Converter (NDCC) improves resource utilization to <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$95.6 \sim 98.7 \%$</tex>. LLM-CIM achieves 126.7TOPS/W energy efficiency on LLaMA2-7B, with up to <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$5.36 x$</tex> energy saving over SOTA CIMs.
Authors
- Shaojun Wei (ORCID: https://orcid.org/0000-0001-5117-7920)
- Fengbin Tu (ORCID: https://orcid.org/0000-0003-2228-8829)
- Longke Yan (ORCID: https://orcid.org/0009-0009-1645-8829)
- Shouyi Yin (ORCID: https://orcid.org/0000-0003-2309-572X)
- Yang Wang (ORCID: https://orcid.org/0000-0001-5057-1870)
- Yang Hu (ORCID: https://orcid.org/0000-0001-6942-4395)
- Ruiqi Guo (ORCID: https://orcid.org/0000-0003-4729-7385)
- Zhen He (ORCID: https://orcid.org/0000-0001-6302-6556)
- Huiming Han
- Yiqi Wang (ORCID: https://orcid.org/0000-0002-9657-3617)
- Zihan Wu (ORCID: https://orcid.org/0000-0002-5858-8850)
Institutions
- Hong Kong University of Science and Technology (HK)
- Tsinghua University (CN)
Publication Details
- Published
- 2025-06-08
- DOI
- https://doi.org/10.23919/vlsitechnologyandcir65189.2025.11074939
- Citations
- 2
- Primary Topic
- Advanced Data Storage Technologies
- Type
- article
- Field-Weighted Citation Impact
- 2.97
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China