跳到正文
AGI Hunt· burny_tech·· 2 小时前AI 评分41

Softmax 墙:低精度 attention 的瓶颈已从矩阵乘法转向超越函数与片上内存搬运

Softmax 墙:低精度注意力已不再受矩阵乘法瓶颈

AI 导读

低精度 attention 在 FP8/FP4 下已受 Softmax 超越函数、向量运算和片上内存搬运限制,而非矩阵乘法。AMD MI355X 上两个 attention GEMM 仅需 1024 cycles,但精确 FP8 softmax 耗时 896 cycles;分析认为,粗粒度 token 级稀疏才是可行方向。

来源:AGI Hunt · agihunt.info