察元AI教程库 · 文章 · 部署

POD-Attention:解锁完整的预填充-解码重叠,实现更快的LLM推理

LLM推理中的每个请求都经过两个阶段:计算密集的prefill阶段和内存带宽密集的decode阶段。为了提高GPU利用率,最近的系统采用混合批处理,将不同请求的prefill和decode阶段组合到同一个批次中。这种方法优化了线性操作。

作者:Aditya K Kamath, Ramya Prabhu, Jayashree Mohan · 语言:en

前往原址学习

简介

LLM推理中的每个请求都经过两个阶段:计算密集的prefill阶段和内存带宽密集的decode阶段。为了提高GPU利用率,最近的系统采用混合批处理,将不同请求的prefill和decode阶段组合到同一个批次中。这种方法优化了线性操作。

返回教程库 · 察元AI产品家族