PyPTX 在B300上超越FA4

A_K_Nain · x · 2026-07-16

PyPTX Flash Attention 在 B300 上被拿来和 FlashAttention 4 对比,引用内容称它在各项指标上都更快:BF16 最高提升约 7.1%,FP8 最高提升约 6.9%,并且使用严格的配对计时方法。 同时,引用帖还声称 PyPTX 的数值精度比 FA4 更好。发帖者对此表示“在发力”,暗示后续还可能有进一步优化空间。

原文链接 →