FeatherAttn: Attention backward kernel on Strix Halo, 4x fast as FlashAttn
https://github.com/woct0rdho/feather-attn On RDNA3/3.5 GPUs, FlashAttn wires to Triton kernels in AITER (AMD's kernel collection). The fwd kernel is fast enough, but it seems no one has optimized the bwd kernel. I've made a bwd kernel with a better designed pipeline. When tested on Strix Halo, it's…
Read the full story at r/StableDiffusion ↗
Timeline · 1 report
- 2026-10-30 18:20 · r/StableDiffusion
FeatherAttn: Attention backward kernel on Strix Halo, 4x fast as FlashAttn