SC22 Proceedings

The International Conference for High Performance Computing, Networking, Storage, and Analysis

Technical Papers Archive

AlphaSparse: Generating High Performance SpMV Codes Directly from Sparse Matrices

Authors: Zhen Du (Institute of Computing Technology, Chinese Academy of Sciences; Chinese Academy of Sciences); Jiajia Li (North Carolina State University); and Yinshan Wang, Xueqi Li, Guangming Tan, and Ninghui Sun (Institute of Computing Technology, Chinese Academy of Sciences)

Abstract: Sparse Matrix-Vector multiplication (SpMV) is an important computational kernel. Tens of sparse matrix formats and implementations have been designed to speed up SpMV performance. We develop AlphaSparse. It goes beyond the scopes of human-designed artificial formats and traditional auto-tuners subject to prior existing artificial format(s) and implementation(s), by automatically creating new machine-designed formats and SpMV kernel implementations entirely from the knowledge of input sparsity patterns and hardware architectures. Based on our proposed Operator Graph that expresses the path of SpMV code design, it takes an arbitrary sparse matrix as input while outputting the machine-designed format and SpMV implementation that achieve high performance. By extensively evaluating 843 matrices from SuiteSparse Matrix Collection, AlphaSparse achieves performance improvement by up to 22.2 times (3.2 times on average) compared to state-of-the-art five artificial formats and up to 2.8 times (1.5 times on average) over the up-to-date implementation of traditional auto-tuning.

Back to Technical Papers Archive Listing