-
题名基于多面体模型的矩阵乘法向量代码生成
被引量：4
- 1
-
-
作者
王博漾
庞建民
徐金龙
赵捷
陶小涵
朱雨
-
机构
郑州大学网络空间安全学院
数学工程与先进计算国家重点实验室(信息工程大学)
-
出处
《计算机科学》
CSCD
北大核心
2022年第10期44-51,共8页
-
基金
国家自然科学基金区域创新发展联合基金(U20A20226)。
-
文摘
矩阵乘法是众多科学计算的核心,而向量化编程是提升其性能的主要手段之一。针对现有的向量化优化往往存在需要手工进行调优以及与硬件结构映射的问题,基于多面体编译器PPCG,在多面体模型中引入向量代码生成框架,提出了基于多面体模型的矩阵乘法向量代码生成框架。通过对矩阵乘法的向量化方案进行收益分析来确定向量化方案,指导应用框架的代码生成,基于该代码生成框架,有利于矩阵乘法的向量化快速优化。选取13个规模在64×64×64到1 024×1 024×1 024之间的矩阵乘法用例进行实验,结果表明,该框架能够正确生成向量化代码,与基础编译器ICC的自动向量化功能相比,应用该框架生成的向量化代码最高获得了5.09倍的加速和3.39倍的平均加速。
-
关键词
矩阵乘法
多面体模型
向量化
调度变换
代码生成
-
Keywords
Matrix multiplication
Polyhedron model
Vectorized
Scheduling transformation
Code generation
-
分类号
TP312
[自动化与计算机技术—计算机软件与理论]
-
