arXiv cs.AI·9 小时前★ 3

提出 Fused-Softmax,量化注意力最高提速 3.2 倍

原文标题:Fused-Softmax for efficient quantized attention

重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。

AI 智能摘要

研究提出免训练的 Fused-,将 主要运算折叠进 GEMM,把负载从向量单元转移到矩阵核心。在 CDNA 4 架构与 MI355X GPU 上,MXFP8/6/4 内核较 提速 2.1 至 3.2 倍,且额外误差低于量化误差下限。内核已开源进 AITER 库,并支撑 AMD v6.1 提交。

深读 · 结论先行AI 生成展开看要点与影响 ↓

AMD Fused- 把 softmax 折进注意力 GEMM,MXFP4 内核较 注意力提速 3.2 倍

  • •免训练算法,将 softmax 主要运算折叠进注意力 GEMM,负载从向量单元转移到矩阵核心
  • •在 CDNA 4 架构与 MI355X 上,MXFP8/6/4 内核达约 3.0/4.0/4.6 PFLOP/s,较 BF16 注意力提速 2.1/2.8/3.2 倍
  • •额外误差低于量化 GEMM 操作数本身的误差下限
  • •Wan2.2-T2V-A14B 上混合 MXFP6/4 调度端到端提速最高 2.05 倍;内核已开源进 AITER,并支撑 AMD MLPerf Inference v6.1 提交,同硬件较上轮快 1.7 倍

为什么重要:softmax 是低精度矩阵吞吐增长下的注意力瓶颈,该方案让注意力吃到矩阵核心的性能红利,对 AMD GPU 上的推理部署成本有直接意义。

以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。

阅读原文 +0
AI 处理过程展开/收起
原始标题:Fused-Softmax for efficient quantized attention
AI 中文标题:AMD 提出 Fused-Softmax,量化注意力最高提速 3.2 倍
来源:arXiv cs.AI · 发布于 9 小时前
分类:算力芯片 · AI 重要度:★ 3(3)
标签:AMD / 注意力机制 / 低精度量化 / 推理加速 / MLPerf
关键词:AMD Fused-Softmax / 量化注意力 / AMD MI355X / 低精度推理 / AITER 库
AI 处理:2026/11/9 15:03:00 · 模型 deepseek-chat · 规则版本 2026-10-21.1
AI 生成以上内容由 AI 自动生成:中文标题、摘要、分类与重要度;同一事件的多家报道会自动归并展示。 内容仅供参考,可能存在偏差,请以原文为准, 不构成任何投资或专业建议。
版权归原作者与原发布平台所有;若涉及侵权或信息错误,请在举报与侵权投诉页反馈,我们会及时处理。
每天 09:30,3 条 AI 圈真正重要的变化,免费订阅 →

相关推荐

36氪·12 天前★ 5

OpenAI 据悉计划采购谷歌 TPU 芯片,降低对英伟达依赖

AI 摘要据报道,OpenAI 正计划采购谷歌自研的张量处理单元(TPU),以扩大算力规模并实现芯片供应多元化,双方尚未敲定最终条款。OpenAI 同时在推进与博通合作、代号 Jalapeno 的自研 AI 芯片。若交易落地,将直接冲击英伟达在 AI 加速芯片市场的主导地位,并重塑云厂商与大模型公司的算力供应格局。

#算力芯片
Hacker News – AI·58 天前★ 5

阿联酋因伊朗袭击调整 5GW AI 数据中心计划

AI 摘要据路透社报道,在遭受伊朗袭击后,阿联酋修订了其 5GW AI 数据中心建设计划。该事件凸显地缘政治冲突对中东地区大规模 AI 基础设施投资的直接冲击,可能影响全球算力布局与供应链。

#算力芯片
The Verge AI·74 天前★ 5

英伟达预计单季营收达1080亿美元

AI 摘要英伟达预测未来几个月内营收将达到1080亿美元,有望成为继亚马逊、苹果和Alphabet之后又一家季度营收超千亿美元的公司。其数据中心营收同比增长一倍以上,利润也大幅增长。

#算力芯片
36氪·1 小时前★ 4

谷歌推出两款新版 TPU 芯片,加速追赶英伟达

AI 摘要谷歌推出两款全新版本张量处理单元(TPU),用于训练和运行 AI 软件,并支持机密计算与更广范围信息调取,适配 AI 智能体趋势。谷歌芯片业务以往每两年推出一款新 AI 处理器,今年一次性发布两款,并将于周一官宣 8tx、8i Max 细分变体。芯片负责人阿明·瓦德特称四款产品分别适配不同需求,强调专用化方向。

#算力芯片
内容有误或摘要不准?告诉我们
AMD 提出 Fused-Softmax,量化注意力最高提速 3.2 倍 - 智讯 AI News