跳到正文
Modal 官方工程博客· Charles Frye·· 2026-06-11AI 评分45

Modal 优化 FlashAttention-4 内核以提升 LLM 推理性能

Making FlashAttention-4 faster for inference

AI 导读

Modal 通过调整并行策略和支持不规则内存访问,优化了 FlashAttention-4 内核以适配大语言模型推理。新增 FP8 输入支持使吞吐量较 bf16 基线提升最高 1.16x,并显著减小 KV cache 体积。这些改进针对 decode-heavy 工作负载,利用 CuTe DSL 实现了更高效的低精度计算与长上下文处理。

来源:Modal 官方工程博客 · modal.com