文章主要内容总结本文聚焦边缘设备上大语言模型（LLM）推理的挑战，提出了一种基于混合 systolic阵列（HSA）架构的边缘LLM推理加速器，旨在优化推理过程中计算密集的prefill阶段和内存受限的decode阶段的效率。背景与挑战：边缘LLM推理需兼顾安全性、低延迟和成本效益，但面临内存容量有限、功耗约束严格、decode阶段内存带宽窄导致性能受限等问题。LLM推理的prefill阶段以矩阵-矩阵乘法（MMM）为主，decode阶段以矩阵-向量乘法（MVM）为主，且decode阶段占总延迟的80%以上，外部内存访问（EMA）是能耗的主要来源。解决方案：提出混合 systolic阵列（HSA）架构，结合传统systolic阵列和向量单元的优势，在prefill阶段实现高能效，在decode阶段保持高硬件利用率和面积效率。采用MXINT4权重量化方案减少EMA，并设计适配HSA的数据流，确保低量化开销和100%硬件利用率，同时精度损失极小。优化根 mean square 归一化（RMSNorm）和旋转位置编码（RoPE）单元，减少其延迟、面积和内存访问开销，支持端到端推理。实验结果：在1.3B参数的RetNet模型上，长输入/长输出场景下实现247/117 token/s/mm²的性能，较现有方案提升2.45×/13.5×，同时在token生成中保持优异的能效。创新
