从三层循环到分块、NEON、AMX 的矩阵乘法优化阶梯

矩阵乘法优化:从三层循环到 Apple Silicon

在 Mac mini M2 Pro 上把朴素 GEMM 从 2.5 GFLOPS 推到 Accelerate 的 2.7 TFLOPS。每一步先看数据怎么走,再改代码,最后对照这颗芯片的存储层次和 AMX。

九月 6, 2026 · 9 分钟 · 4224 字 · Marshall Liu