#39119 [ROCm] Align AiterFlashAttentionImpl attn_type check with backend
作者 Bortlesboat · 合并时间 2026-04-15 01:36
修复ROCm平台AiterFlashAttentionImpl中attn_type检查与后端不一致的问题,防止跨注意力错误计算。
该PR值得快速浏览,重点关注attn_type检查的逻辑对齐和错误信息的改进。对于关注ROCm平台注意力后端实现的开发者,这是一个重要的防御性修复,展示了后端契约与实现类保持一致的重要性。
参与讨论