研究人员开发了 ZYFA,这是一组用于 AMD Instinct MI355X 加速器的新预填充内核,旨在提高 FlashAttention 的速度和准确性。ZYFA 采用了三种关键技术:FP8 概率的直接计算、用于重叠 softmax 和矩阵乘法的乒乓调度,以及用于归一化的缩放方法。这些优化使得吞吐量比现有的 AITER 内核提高了 1.16-1.76 倍,并显著减少了数值误差,使其更接近 Composable Kernel 和 Triton 的水平。 AI
影响 提高了核心注意力机制的效率,有可能在特定硬件上加速大型语言模型的训练和推理。
排序理由 学术论文,详细介绍了用于核心人工智能模型组件硬件加速的新颖优化技术。[lever_c_demoted from research: ic=1 ai=1.0]
- AMD
- AMD Instinct MI355X
- Blackwell
- Composable Kernel
- DeepSeek
- FlashAttention
- FP8
- Hopper
- Nvidia
- Triton
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →