Optimizing Sparse Linear Algebra Through Automatic Format Selection and Machine Learning

Chris Stylianou*, Michele Weiland

*Corresponding author for this work

Research output: Contribution to conferencePaperpeer-review

Abstract / Description of output

Sparse matrices are an integral part of scientific simulations. As hardware evolves new sparse matrix storage formats are proposed aiming to exploit optimizations specific to the new hardware. In the era of heterogeneous computing, users often are required to use multiple formats for their applications to remain optimal across the different available hardware, resulting in larger development times and maintenance overhead. A potential solution to this problem is the use of a lightweight auto-tuner driven by Machine Learning (ML) that would select for the user an optimal format from a pool of available formats that will match the characteristics of the sparsity pattern, target hardware and operation to execute.
In this paper, we introduce Morpheus-Oracle, a library that provides a lightweight ML auto-tuner capable of accurately predicting the optimal format across multiple backends, targeting the major HPC architectures aiming to eliminate any format selection input by the end-user. From more than 2000 real- life matrices, we achieve an average classification accuracy and balanced accuracy of 92.63% and 80.22% respectively across the available systems. The adoption of the auto-tuner results in average speedup of 1.1× on CPUs and 1.5× to 8× on NVIDIA and AMD GPUs, with maximum speedups reaching up to 7× and 1000× respectively.
Original languageEnglish
Publication statusPublished - 4 Aug 2023
EventInternational Workshop on Automatic Performance Tuning - Florida, Petersburg, United States
Duration: 19 May 2023 → …


WorkshopInternational Workshop on Automatic Performance Tuning
Abbreviated titleiWAPT
Country/TerritoryUnited States
Period19/05/23 → …


Dive into the research topics of 'Optimizing Sparse Linear Algebra Through Automatic Format Selection and Machine Learning'. Together they form a unique fingerprint.

Cite this