[RFC] FA4 — head_dim=256 & head_dim=512

作者: Johnsonms创建于 2026年4月13日更新于 2026年9月14日

概述

内容来源: Dao-AILab/flash-attention