跳到正文
Reddit · r/StableDiffusion· /u/woct0rdho·· 4 小时前AI 评分20

FeatherAttn:Strix Halo 上的注意力反向 kernel,比 FlashAttn 快 4 倍

FeatherAttn: Attention backward kernel on Strix Halo, 4x fast as FlashAttn

AI 导读

开发者 woct0rdho 为 RDNA3/3.5 GPU 推出注意力反向 kernel FeatherAttn,在 Strix Halo 上 head dim 128 时比 FlashAttn 快 4 倍,head dim 64 时快 1.3 倍。

打开原文reddit.com

来源:Reddit · r/StableDiffusion · reddit.com