arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.03142cs.ARcs.LGcs.PFcs.PL
收藏

Power ISA(TM)处理器的矩阵数学工具

A matrix math facility for Power ISA(TM) processors

International Business Machines Corporation(国际商业机器公司)
José E. Moreira, Kit Barton, Steven Battle, Peter Bergner, Ramon Bertran, Puneeth Bhat, Pedro Caldeira, David Edelsohn, Gordon Fossum, Brad Frey, Nemanja Ivanov… 展开作者

José E. Moreira, Kit Barton, Steven Battle, Peter Bergner, Ramon Bertran, Puneeth Bhat, Pedro Caldeira, David Edelsohn, Gordon Fossum, Brad Frey, Nemanja Ivanovic, Chip Kerchner, Vincent Lim, Shakti Kapoor, Tulio Machado Filho, Silvia Melitta Mueller, Brett Olsson, Satish Sadasivam, Baptiste Saleil, Bill Schmidt, Rajalakshmi Srinivasaraghavan, Shricharan Srivatsan, Brian Thompto, Andreas Wagner, Nelson Wu

摘要:

Power ISA(TM)3.1版引入了一系列新的矩阵数学指令,统称为矩阵乘法辅助(MMA)工具。该工具中的指令在小矩阵上实现数值线性代数运算,旨在加速计算密集型内核,例如矩阵乘法,卷积和离散傅立叶变换。这些指令在未来的POWER10处理器中实现了高吞吐量数学引擎的功耗和面积效率。在恒定频率下,每个核心的性能比上一代POWER9处理器高4倍。我们还提倡使用编译器内置程序作为利用这些指令的首选方式,我们通过案例研究来说明,包括矩阵乘法和卷积。

英文摘要:

Power ISA(TM) Version 3.1 has introduced a new family of matrix math instructions, collectively known as the Matrix-Multiply Assist (MMA) facility. The instructions in this facility implement numerical linear algebra operations on small matrices and are meant to accelerate computation-intensive kernels, such as matrix multiplication, convolution and discrete Fourier transform. These instructions have led to a power- and area-efficient implementation of a high throughput math engine in the future POWER10 processor. Performance per core is 4 times better, at constant frequency, than the previous generation POWER9 processor. We also advocate the use of compiler built-ins as the preferred way of leveraging these instructions, which we illustrate through case studies covering matrix multiplication and convolution.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑