GEMM 量化术语百科词条
词条类别:量化基础概念(量化模式) 英文名称:GEMM(General Matrix Multiply) 应用领域:线性层计算、推理加速
1. 概述
GEMM(General Matrix Multiply,通用矩阵乘法)即矩阵乘 C=A×BC=A\times B,是 Transformer 中线性层计算 Y=X⋅WY=X\cdot W 的数学形式。量化使 GEMM 能以整数/低精度输入运行,是量化的主要计算收益来源。
2. 词条介绍
定义
GEMM 是 C=A×BC=A\times B 形式的矩阵乘法。Transformer 中占比最高的计算就是 GEMM:Q/K/V 投影、注意力输出投影、MLP 三层都是线性层,各自做一次矩阵乘。
整数 GEMM
整数 GEMM(Integer GEMM)指输入为 INT8/INT4 等低精度整数、乘累加在整数域进行的矩阵乘:权重与激活都量化成整数后,全程整数计算,结果再乘回 scale 还原。整数 GEMM 可调用专用低精度算子,是量化的主要计算收益来源。
若只有一侧量化(如 W8A16 静态量化),则需把整数反量化回浮点再做浮点 GEMM,没有整数 GEMM 的加速。
与量化的关系
- 前提是量化与反量化:权重与激活都量化成整数,才能进入整数域计算。
- 数据类型的选取(INT8、FP8、MXFP8/MXFP4)决定整数/低精度 GEMM 的硬件算子与精度表现。
3. 关联流程
4. 关联词条
5. 参考文档
- 《量化模式》