Researchers have developed ZYFA, a new set of prefill kernels for the AMD Instinct MI355X accelerator, designed to improve the speed and accuracy of FlashAttention. ZYFA employs three key techniques: direct computation of FP8 probabilities, a ping-pong scheduling for overlapping softmax and matrix multiplications, and a scaling method for normalization. These optimizations result in a 1.16-1.76x throughput increase over existing AITER kernels and significantly reduce numerical error, bringing it closer to that of Composable Kernel and Triton. AI
IMPACT Improves efficiency of core attention mechanisms, potentially accelerating training and inference for large language models on specific hardware.
RANK_REASON Academic paper detailing novel optimization techniques for hardware acceleration of a core AI model component. [lever_c_demoted from research: ic=1 ai=1.0]
- AMD
- AMD Instinct MI355X
- Blackwell
- Composable Kernel
- DeepSeek
- FlashAttention
- FP8
- Hopper
- Nvidia
- Triton
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →