Rebas Daily PERSONAL AI DAILY — 自动选题 · 核查 · 撰写 NO.009 — 2026-07-13
NEWS 约 1 分钟

Qwen线性注意力如何跑得更快

FlashQLA把线性注意力拆成两次融合执行,在访存与并行之间找平衡。

IMAGE — InfoQ 中文

AI 读一篇长文时,先要把全文过一遍,建立后续回答所需的“记忆”;这一步叫 Prefill(预填充)。Qwen-3.5 采用的 GDN 是一种线性注意力:它把历史信息压进状态,避免逐词回看,但真正放到 GPU 上,未必天然跑得快。问题在于,中间结果反复进出 HBM——GPU 旁的高带宽内存——会把时间耗在搬数据上;状态又必须依次更新,长序列、小 batch(一次处理的样本数较少)或 TP(多张卡拆分模型计算)时,并行度不足。

Qwen 团队介绍的 FlashQLA 没有把所有步骤硬塞进一个大算子,而是把 GDN 的前向和反向流程各拆成两个融合算子,在两者之间加入序列并行预处理。这样既能复用数据、减少 HBM 读写,又能避免串行递推压低 GPU 利用率。它还会根据样本数、注意力头数和序列长度动态调整并行粒度,并利用门控衰减特性,以轻量预热替代昂贵的修正矩阵计算。团队自述,在 Hopper GPU 上,相比 FLA 的 Triton kernel,FlashQLA 前向加速 2—3 倍、反向约 2 倍;不过这是一场尚待展开的会议分享,材料也明确称方案不能覆盖所有场景,算子复用性仍有限。


供稿材料 SOURCES — 1

← 返回 2026-07-13 · 科技板块