• Home
  • Search
  • Optimization of Sparse Matrix Computation for Algebraic Multigrid on GPUs
  • Cite Icon3
  • https://doi.org/10.1145/3664924Copy DOI Icon

Optimization of Sparse Matrix Computation for Algebraic Multigrid on GPUs

Show More
  • Abstract
  • Highlights & Summary
  • PDF
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

AMG is one of the most efficient and widely used methods for solving sparse linear systems. The computational process of AMG mainly consists of a series of iterative calculations of generalized sparse matrix-matrix multiplication (SpGEMM) and sparse matrix-vector multiplication (SpMV). Optimizing these sparse matrix calculations is crucial for accelerating solving linear systems. In this paper, we first focus on optimizing the SpGEMM algorithm in AmgX, a popular AMG library for GPUs. We propose a new algorithm called SpGEMM-upper, which achieves an average speedup of 2.02× on Tesla V100 and 1.96× on RTX 3090 against the original algorithm. Next, through experimental investigation, we conclude that no single SpGEMM library or algorithm performs optimally for most sparse matrices, and the same holds true for SpMV. Therefore, we build machine learning-based models to predict the optimal SpGEMM and SpMV used in the AMG calculation process. Finally, we integrate the prediction models, SpGEMM-upper, and other selected algorithms into a framework for adaptive sparse matrix computation in AMG. Our experimental results prove that the framework achieves promising performance improvements on the test set.

Loading PDF

Similar Papers
  • Research Article
  • Citations1

SpChar: Characterizing the sparse puzzle via decision trees

  • Jun 17, 2024
  • Journal of Parallel and Distributed Computing
  • Francesco Sgherzi +4
  • Research Article
  • Citations356

Sparsity: Optimization Framework for Sparse Matrix Kernels

  • Feb 01, 2004
  • The International Journal of High Performance Computing Applications
  • Eun-Jin Im +2
  • Conference Article
  • Citations3

Energy consumption optimization of the Total-FETI solver and BLAS routines by changing the CPU frequency

  • Jul 01, 2016
  • David Horak +4
  • Research Article
  • Citations6

Heterogeneous sparse matrix–vector multiplication via compressed sparse row format

  • Jan 20, 2023
  • Parallel Computing
  • Phillip Allen Lane +1
  • Conference Article
  • Citations8

Optimizing Sparse Matrix Vector Multiplication Using Diagonal Storage Matrix Format

  • Sep 01, 2010
  • Liang Yuan +3
  • Research Article
  • Citations33

SMAT

  • Jun 16, 2013
  • ACM SIGPLAN Notices
  • Jiajia Li +3
  • Book Chapter
  • Citations2

Sparse matrices: storage schemes and simple operations

  • Jan 26, 2017
  • I S Duff +2
  • Book Chapter
  • Citations7

Optimizing LOBPCG: Sparse Matrix Loop and Data Transformations in Action

  • Jan 01, 2017
  • Khalid Ahmad +2
  • Conference Article
  • Citations39

Performance Analysis and Optimization of Sparse Matrix-Vector Multiplication on Modern Multi- and Many-Core Processors

  • Aug 01, 2017
  • Athena Elafrou +2
  • Conference Article
  • Citations73

Sparse Matrix Format Selection with Multiclass SVM for SpMV on GPU

  • Aug 01, 2016
  • Akrem Benatia +3
  • Conference Article
  • Citations45

Structural Agnostic SpMV: Adapting CSR-Adaptive for Irregular Matrices

  • Dec 01, 2015
  • Mayank Daga +1
  • Conference Article
  • Citations42

Adaptive Optimization of Sparse Matrix-Vector Multiplication on Emerging Many-Core Architectures

  • Jun 01, 2018
  • Shizhao Chen +4
  • Research Article
  • Citations5

A sparse matrix–vector multiplication based algorithm for accurate density matrix computations on systems of millions of atoms

  • Feb 15, 2018
  • Computer Physics Communications
  • Purnima Ghale +1
  • Conference Article
  • Citations21

Cache-aware sparse matrix formats for Kepler GPU

  • Dec 01, 2014
  • Yusuke Nagasaka +2
  • Conference Article
  • Citations3

Column-Segmented Sparse Matrix-Matrix Multiplication on Multicore CPUs

  • Dec 01, 2021
  • Xiaojing An +1
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.