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.

The Chunks and Tasks Matrix Library 2.0

Parallelization and scalability analysis of inverse factorization using the Chunks and Tasks programming model

Loading Large Sparse Matrices Stored in Files in the Adaptive-Blocking Hierarchical Storage Format

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

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

Parallel Cholesky Factorization for Banded Matrices using OpenMP Tasks

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

Computing Sparse Tensor Decompositions via Chapel and C++/MPI Interoperability without Intermediate I/O

Distributed-Memory Parallel Algorithms for Sparse Matrix and Sparse Tall-and-Skinny Matrix Multiplication

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

Scheduling Trees of Malleable Tasks for Sparse Linear Algebra

Automatic Tuning of Sparse Matrix-Vector Multiplication on Multicore Clusters.

FMMTL: FMM Template Library A Generalized Framework for Kernel Matrices

Cache Blocking of Distributed-Memory Parallel Matrix Power Kernels

Experiences Porting Distributed Applications to Asynchronous Tasks: A Multidimensional FFT Case-study

Shared Memory Parallelization of MTTKRP for Dense Tensors

Scale-Free Sparse Matrix-Vector Multiplication on Many-Core Architectures

A More Scalable Sparse Dynamic Data Exchange

Accelerating Task-based Iterative Applications

Task-D: A Task Based Programming Framework for Distributed System.

GHOST: Building blocks for high performance sparse linear algebra on heterogeneous systems