A new attention backward kernel called FeatherAttn has been developed for RDNA3/3.5 GPUs, offering significant speedups over existing implementations like FlashAttn. When tested on Strix Halo hardware, FeatherAttn demonstrated a 4x speed increase for a head dimension of 128 and a 1.3x increase for a head dimension of 64. This optimization is particularly relevant for recent AI models that utilize a head dimension of 128, such as Anima, Krea 2, MOSS, and H3, and has been shown to accelerate LoRA training by up to 40%. AI
IMPACT Accelerates AI model training and LoRA fine-tuning on compatible hardware.
RANK_REASON New kernel optimization for AI model training on specific hardware. [lever_c_demoted from research: ic=1 ai=0.7]
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →