计算物理2024,Vol.41Issue(1) :33-39.DOI:10.19596/j.cnki.1001-246x.8763

稀疏矩阵向量乘的自动调优

Auto-tuning for Sparse Matrix-vector Multiplication

杜臻 谭光明
计算物理2024,Vol.41Issue(1) :33-39.DOI:10.19596/j.cnki.1001-246x.8763

稀疏矩阵向量乘的自动调优

Auto-tuning for Sparse Matrix-vector Multiplication

杜臻 1谭光明2
扫码查看

作者信息

  • 1. 中国科学院大学计算机科学与技术学院,北京 101408;中国科学院计算技术研究所,北京 100190
  • 2. 中国科学院计算技术研究所,北京 100190
  • 折叠

摘要

分析稀疏矩阵向量乘(SpMV)程序优化的难点,介绍两个自动调优的代表性工作:基于预实现模板的SMAT和从头设计程序的 AlphaSparse.详细介绍了它们的设计思路、实现细节、测试结果以及各自的优缺点.最后,对 SpMV自动调优的发展趋势进行了分析和预测.

Abstract

SpMV(sparse matrix-vector multiplication)is a widely used kernel in scientific computing.Since the performance of specific SpMV program is closely related to the distribution of non-zero elements in sparse matrices,there is no universal SpMV program design that can achieve high performance in all matrices.Therefore,auto-tuning has become a popular method for high SpMV performance.This paper analyzes the difficulties in optimizing SpMV and introduces two representative works of auto-tuning:SMAT,which is based on pre-implemented templates and AlphaSparse which designs SpMV programs from scratch.This paper introduces their designs,implementations,test results,advantages,and disadvantages.Finally,the trend of SpMV auto-tuning is analyzed and predicted.

关键词

高性能科学计算/稀疏矩阵/自动调优/稀疏矩阵向量乘

Key words

high-performance scientific computing/sparse matrix/auto-tuning/sparse matrix-vector multiplication

引用本文复制引用

基金项目

国家自然科学基金杰出青年基金项目(T2125013)

出版年

2024
计算物理
中国核学会

计算物理

CSTPCD北大核心
影响因子:0.366
ISSN:1001-246X
参考文献量19
段落导航相关论文