矩阵乘协处理器上BLAS level-3运算的设计

• 作者: 原昊 吴东 张昆 贾迅 钱磊
• 来源期刊: 计算机工程与科学
• 学科: 工学
• 关键词: 线性阵列 矩阵乘 协处理器 BLAS level-3
• 年，卷（期）: 2020,（11）
• 所属期刊栏目: 高性能计算
• 页码范围: 1913-1921
• 页数: 9页
• 分类号: TP303
• 语种: 中文
• DOI: 10.3969/j.issn.1007-130X.2020.11.001

摘要

BLAS level-3运算的计算复杂度较高,其往往成为应用的性能瓶颈.采用线性阵列结构的矩阵乘协处理器可实现高性能、高效的矩阵乘运算.在矩阵乘协处理器上高效实现BLAS level-3运算,对大规模科学与工程仿真应用的计算加速至关重要.以矩阵乘为核心运算,结合线性阵列的结构特点,提出了矩阵乘协处理器上BLAS level-3运算的设计,并构建了相应的性能分析模型.实验结果表明,矩阵乘协处理器上SYMM、SYRK和TRMM运算的计算效率分别达到了99％,98％ 和80％,与SW26010和NVIDIA V100 GPU上矩阵运算的计算效率相比,最高提升了31％.