FlashMask-3 发布:Hopper 内核加速长上下文分布式注意力训练

2026 年 10 月 13 日,Hugging Face Papers 收录的论文提出 FlashMask-3,针对 Hopper GPU 为长上下文训练的注意力掩码提供分布式加速方案,在摘要所列实验中最多将吞吐量提升至基线数倍。

一句话看懂:2026 年 10 月 13 日,Hugging Face Papers 收录的论文提出 FlashMask-3,针对 Hopper GPU 为长上下文训练的注意力掩码提供分布式加速方案,在摘要所列实验中最多将吞吐量提升至基线数倍。

事件核心:发生了什么

论文指出,长上下文大模型与多模态训练中,文档、prefix-LM、滑动窗口、块稀疏等结构化注意力掩码会与多节点上下文并行叠加,成为主要瓶颈。前代 FlashMask 用四组查询轴边界向量把掩码存储从 O(n²) 压缩到 O(n),但只面向 Ampere,既不利用 Hopper 的 TMA/WGMMA 流水线,也不支持分布式。FlashMask-3 补齐这两块:新增三角色 warp 特化的 Hopper 内核,并给出零通信掩码定位、稀疏感知负载均衡(HCC-IPO)和掩码驱动的计算—通信重叠与通信剪枝。按摘要,在单张 H100、4K–128K 上下文的 12 种掩码上,其 TFLOPs 较 FlashMask、FlexAttention、MagiAttention 分别高 45–141%、34–71%、2–44%;上下文并行度 4–32 时,计算通信重叠效率为 31–75%。

为什么重要

长上下文正成为大模型训练与推理的基础能力,而掩码与上下文并行的组合长期依赖碎片化实现。FlashMask-3 试图把此前分属“单卡内核”和“多卡调度”的工作合并到一个掩码感知栈中,其零通信掩码定位可降低多卡同步开销,拓扑感知分层路由则针对 all-gather 通信做优化。若后续实验能被复现,这类优化对使用 H100 集群训练大模型的团队有直接吸引力。

对用户/开发者/创作者的影响

对开发者而言,关注点在于其代码是否开源、能否接入 FlashAttention-3 生态以及 Megatron 类训练框架;对使用长上下文 API 或产品的用户,这类底层优化若落地,可能体现为更低的长文档处理延迟和成本,但论文摘要并不等于产品上线。企业采购与算力团队可留意其在高并行度下宣称的吞吐提升是否在自有多模态和 MoE 训练中成立。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一是后续公开全文与代码是否释放,HCC-IPO 等组件能否复现;二是 118B-A14B MoE 在 128 卡 128K 训练中相对 Dual Chunk 基线提升 7.3–11.0% 的结论,需在更多框架和硬件配置上验证;三是 FlashAttention、FlexAttention、MagiAttention 等竞品是否跟进掩码感知的分布式路线。

来源:Hugging Face Papers

celebrityanime
celebrityanime
文章: 29324

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注