一篇新论文介绍了 2M 乘法算法,该算法将复数矩阵乘法的成本从三次实数 General Matrix Multiplications (GEMMs) 降低到仅两次。该方法通过利用二次时间的预处理和后处理步骤,适用于具有整数实部和虚部的矩阵。
- 该算法取代了标准的 3M 方法,仅需对复数输入执行两次相同大小的实数 GEMM。
- 对于浮点矩阵,它与 Ozaki-II 方案集成以实现高性能,耗时约为实数 GEMM 的两倍。
- 该工作还推导出了用于对称秩-k 更新 (SYRK/HERK) 的新算法,这些算法内部使用完整的矩形 GEMMs。
该方法为计算复数浮点 GEMMs 提供了一种实用的高性能解决方案,同时为对称秩运算提供了相应的改进。