@Alacritic_Super: LLM推理的最大瓶颈不是算术运算,而是数据移动。一次乘加运算仅需…
摘要
一条科普线程,解释LLM推理的主要瓶颈是数据移动而非计算,并强调了量化、KV缓存优化和FlashAttention等减少内存流量技术的要点。
查看缓存全文
缓存时间: 2026/07/15 19:59
LLM推理的最大瓶颈并非算术运算,而是数据传输。
一次乘加运算的成本非常低:
FP32:约3–5皮焦耳
FP16:约1–2皮焦耳
INT8:约0.2–0.5皮焦耳
但内存中提取数据的成本通常要高得多:
L1缓存:约1–5皮焦耳
L2缓存:约10–30皮焦耳
HBM / 片外DRAM:约300–1000皮焦耳
这为什么重要?
在推理过程中,GPU会反复加载:
模型权重
激活值
KV缓存
中间张量
在许多工作负载中,传输这些字节所消耗的能量和耗时往往超过矩阵乘法本身。
正因如此,现代推理框架专注于减少内存流量:
量化(FP8、INT8、INT4)
KV缓存优化
PagedAttention
FlashAttention
前缀和提示缓存
核融合
算子融合
内存感知调度
权重压缩
这也是GPU持续大力投资以下技术的原因:
更大的片上SRAM
更高的HBM带宽
更智能的缓存层级
更快的互连(NVLink、NVSwitch)
相似文章
@Alacritic_Super: 如果你认真对待LLM推理,学习FlashAttention。它是现代…背后最重要的优化之一。
一条推文推荐在LLM推理中学习FlashAttention,强调其在优化GPU内存流量和加速注意力机制方面的重要性,并附有FlashAttention、FlashAttention-2和FlashAttention-3的GitHub仓库和论文链接。
内存
解释了为什么由于KV缓存随上下文长度和并发用户数扩展,LLM推理越来越受内存带宽限制,以及像vLLM和PagedAttention这样的系统如何提高内存利用率。
@Alacritic_Super: 如果你在构建生产级 LLM 应用,学习 LLM 缓存。缓存可降低延迟、GPU 利用率和 AP…
本文强调了在生产系统中使用 LLM 缓存的重要性,以减少延迟、GPU 利用率和成本,并介绍了 LMCache,这是一个用于可扩展 LLM 推理的开源 KV 缓存管理层。
@_avichawla: LLM推理中的预填充与解码。你是否注意到,LLM的第一个令牌总是需要片刻才出现…
解释LLM推理的两个阶段——预填充和解码,详细说明GPU瓶颈如何从预填充时的计算受限转变为解码时的内存受限,以及KV缓存的重要性。
@divaagurlxw: 如果我想让LLM响应低于一秒,我会研究的推理优化方法:1.KV-Caching 2.Speculative Decoding 3.FlashAtte…
一条推文列出了16种推理优化技术,用于实现低于一秒的LLM响应,包括KV缓存、推测解码、FlashAttention和各种并行化方法。