Strassen-Winograd 矩阵乘法算法的内存高效调度
数学软件
2009-05-18 v5
摘要
我们提出了几种用于 Strassen-Winograd 矩阵乘法算法的新调度方案,它们通过三种不同方式减少了额外内存分配需求:引入少量预加法、覆盖输入矩阵,或使用经典乘法的第一层递归。特别是,我们展示了两种完全原地 (in-place) 调度方案:如果输入矩阵可以被覆盖,则操作次数相同;如果输入矩阵是只读的,则略微增加复杂度主导项的常数。许多此类调度方案是通过基于石子游戏 (pebble game) 的穷举搜索算法的实现而发现的。
引用
@article{arxiv.0707.2347,
title = {Memory efficient scheduling of Strassen-Winograd's matrix multiplication algorithm},
author = {Brice Boyer and Jean-Guillaume Dumas and Clément Pernet and Wei Zhou},
journal= {arXiv preprint arXiv:0707.2347},
year = {2009}
}