修改 MLX 用 FP8 暂存量化权重:M6 上预填充提升 40%
作者修改 MLX 的 QMM 实现,把量化权重从 FP16 暂存改为 FP8,在 Apple Silicon M6 上启用 2 倍速 FP8 矩阵路径,Qwen3-8B 预填充提升约 40%,Qwen3.8-27B 提升约 50%;4-bit affine 权重暂存为 int8 的方案在 M5 及之后也可用。Decode 受带宽限制差异不大,建议保留默认 FP16 暂存。最坏情况困惑度增加低于 1%,代码在实验性 fork(https://github.com/precisit/mlx/tree/staged-8bit-qmm),FP8 改进需 M6 与 macOS 27 及 deployment-target-27 构建。作者披露博客来自其任职的 Precisit。
推荐理由:作者自己改 MLX 的量化矩阵路径并给出实测加速与困惑度数据,方法细节和代码都可复现。