Abstract:Matrix-matrix multiplication is a basic operation in linear algebra and an essential building block for a wide range of algorithms in various scientific fields. Theory and implementation for the dense, square matrix case are well-developed. If matrices are sparse, with application-specific sparsity patterns, the optimal implementation remains an open question. Here, we explore the performance of communication reducing 2.5D algorithms and one-sided MPI communication in the context of linear scaling electronic structure theory. In particular, we extend the DBCSR sparse matrix library, which is the basic building block for linear scaling electronic structure theory and low scaling correlated methods in CP2K. The library is specifically designed to efficiently perform block-sparse matrix-matrix multiplication of matrices with a relatively large occupation. Here, we compare the performance of the original implementation based on Cannon's algorithm and MPI point-to-point communication, with an implementation based on MPI one-sided communications (RMA), in both a 2D and a 2.5D approach. The 2.5D approach trades memory and auxiliary operations for reduced communication, which can lead to a speedup if communication is dominant. The 2.5D algorithm is somewhat easier to implement with one-sided communications. A detailed description of the implementation is provided, also for non ideal processor topologies, since this is important for actual applications. Given the importance of the precise sparsity pattern, and even the actual matrix data, which decides the effective fill-in upon multiplication, the tests are performed within the CP2K package with application benchmarks. Results show a substantial boost in performance for the RMA based 2.5D algorithm, up to 1.80x, which is observed to increase with the number of processes involved in the parallelization.

Sparse Matrix Multiplication On An Associative Processor

On Parallelizing Matrix Multiplication by the Column-Row Method

In-memory Multi-valued Associative Processor

Dedicated Hardware Accelerators for Processing of Sparse Matrices and Vectors: A Survey

Efficient Distributed-Memory Parallel Matrix-Vector Multiplication with Wide or Tall Unstructured Sparse Matrices

Sparse Matrix Multiplication on CAM Based Accelerator

Fast Matrix Multiplication via Compiler-only Layered Data Reorganization and Intrinsic Lowering

A Work-Efficient Parallel Sparse Matrix-Sparse Vector Multiplication Algorithm

Performance Optimization for Sparse A(T)Ax in Parallel on Multicore Cpu

Increasing the Efficiency of Sparse Matrix-Matrix Multiplication with a 2.5D Algorithm and One-Sided MPI

SpArch: Efficient Architecture for Sparse Matrix Multiplication

From array algebra to energy efficiency on GPUs: Data and hardware shapes with dimension-lifting to optimize memory-processor layouts

VerSA: Versatile Systolic Array Architecture for Sparse and Dense Matrix Multiplications

A sparse matrix vector multiplication accelerator based on high-bandwidth memory

Configurable sparse matrix - matrix multiplication accelerator on FPGA: A systematic design space exploration approach with quantization effects

Computing the sparse matrix vector product using block-based kernels without zero padding on processors with AVX-512 instructions

Implementing High-performance Complex Matrix Multiplication via the 3m and 4m Methods

Parallel Sparse Matrix Multiplication for Preconditioning and SSTA on a Many-Core Architecture

DeMM: A Decoupled Matrix Multiplication Engine Supporting Relaxed Structured Sparsity