基于多面体模型的矩阵乘法向量代码生成

摘要：

矩阵乘法是众多科学计算的核心,而向量化编程是提升其性能的主要手段之一.针对现有的向量化优化往往存在需要手工进行调优以及与硬件结构映射的问题,基于多面体编译器PPCG,在多面体模型中引入向量代码生成框架,提出了基于多面体模型的矩阵乘法向量代码生成框架.通过对矩阵乘法的向量化方案进行收益分析来确定向量化方案,指导应用框架的代码生成,基于该代码生成框架,有利于矩阵乘法的向量化快速优化.选取13个规模在64×64×64到1024×1024×1024之间的矩阵乘法用例进行实验,结果表明,该框架能够正确生成向量化代码,与基础编译器ICC的自动向量化功能相比,应用该框架生成的向量化代码最高获得了5.09倍的加速和3.39倍的平均加速.

关键词：矩阵乘法、多面体模型、向量化、调度变换、代码生成

所属期刊栏目：49

分类号：TP312(计算技术、计算机技术)

资助基金：国家自然科学基金U20A20226

在线出版日期：2022-10-20（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：44-51