AMD RDNA 4 架构的 WMMA 内核优化
2025-07-21
AMD RDNA 4 架构的 GPU 通过第三代矩阵核心提升了通用矩阵乘法 (GEMM) 运算性能。本文深入探讨了如何在 HIP 内核中使用 RDNA 4 GPU 的矩阵核心，特别是利用 WMMA (Wave Matrix Multiply Accumulate) 内核进行矩阵运算的优化技巧。文章详细介绍了 WMMA 的工作原理、新的内建函数 `__builtin_amdgcn_wmma_f32_16x16x16_f16_w32_gfx12` 的使用方法以及与 RDNA 3 架构的差异，并提供了一个简单的多层感知器 (MLP) 实现示例，展示如何在 RDNA 4 上高效执行矩阵运算。
硬件
WMMA