CroissAttn 实现 Blackwell 上注意力 FP4 全程运算:Qwen3-8B 吞吐提升1.5倍

TristanThrush · x · 2026-11-07

Alex Rose Jones 介绍 CroissAttn:在 NVIDIA Blackwell 上,注意力全部六个 matmul 都能跑 FP4。在 128K 上下文下,Qwen3-8B 相比 BF16 获得 1.5× 吞吐、节省 55% GPU 显存。转发者 Anupama Dhar 补充:完全用 FP4 训练或 FP4 微调还能提升 FP4 推理的下游精度,即端到端低精度不会牺牲准确率。作者将在 Nebius Paper Club 分享技术细节。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →