FA kernel 数值实现脆弱:大数相减问题不止 FlashAttention

YouJiacheng · x · 2026-10-04

作者在讨论 attention kernel 的数值稳定性实现,指出 FlashAttention kernel 实现是脆弱的:exp(sij - lse) 本质是大数相减的 delta;对方将 lse 分离、用 c (S - m) - log(l) 的方法里 (S - m) 依然是大数相减。

所属事件:FlashAttention 大数相减隐患引热议(2 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →