稀疏矩阵向量乘法运行时数据转换的自动调优方法
分布式、并行与集群计算
2024-07-02 v1 性能
摘要
本文研究了从压缩行存储 (CRS) 到坐标 (COO) 存储以及带有 OpenMP 并行化的 ELL (ELLPACK/ITPACK) 格式的稀疏矩阵数据运行时转换,用于稀疏矩阵向量乘法 (SpMV)。我们提出了一种自动调优 (AT) 方法,利用 - 图,其中绘制了非零元素每行数的派生/平均值 () 与 SpMV 加速度/从 CRS 到 ELL 转换时间之比 ()。实验结果表明,ELL 格式在 Earth Simulator 2 中非常有效。获得 151 倍的 ELL-Row 内并行化格式加速度因子。转换开销也很小,例如 0.01 到 1.0 个 SpMV 时间。此外, - 图可以根据 值来建模转换的有效性。
引用
@article{arxiv.2407.00019,
title = {An Auto-tuning Method for Run-time Data Transformation for Sparse Matrix-Vector Multiplication},
author = {Takahiro Katagiri and Masahiko Sato},
journal= {arXiv preprint arXiv:2407.00019},
year = {2024}
}