A Tuned and Scalable Fast Multipole Method as a Preeminent Algorithm for Exascale Systems

TL;DR

This paper demonstrates a highly optimized and scalable implementation of the fast multipole method (FMM) on multi-core CPUs and supercomputers, showing its potential as a key algorithm for exascale computing.

Contribution

The paper presents extensive performance tuning and scalability analysis of FMM on CPU clusters, including SIMD optimization and OpenMP parallelization, highlighting its suitability for exascale systems.

Findings

Achieved 78% efficiency on 8 threads for 10^7 particles.

Attained 93% strong scaling efficiency on 2048 processes with 10^8 particles.

Handled over 32 billion unknowns in 40 seconds on large-scale supercomputers.

Abstract

Among the algorithms that are likely to play a major role in future exascale computing, the fast multipole method (FMM) appears as a rising star. Our previous recent work showed scaling of an FMM on GPU clusters, with problem sizes in the order of billions of unknowns. That work led to an extremely parallel FMM, scaling to thousands of GPUs or tens of thousands of CPUs. This paper reports on a a campaign of performance tuning and scalability studies using multi-core CPUs, on the Kraken supercomputer. All kernels in the FMM were parallelized using OpenMP, and a test using 10^7 particles randomly distributed in a cube showed 78% efficiency on 8 threads. Tuning of the particle-to-particle kernel using SIMD instructions resulted in 4x speed-up of the overall algorithm on single-core tests with 10^3 - 10^7 particles. Parallel scalability was studied in both strong and weak scaling. The strong scaling test used 10^8 particles and resulted in 93% parallel efficiency on 2048 processes for the non-SIMD code and 54% for the SIMD-optimized code (which was still 2x faster). The weak scaling test used 10^6 particles per process, and resulted in 72% efficiency on 32,768 processes, with the largest calculation taking about 40 seconds to evaluate more than 32 billion unknowns. This work builds up evidence for our view that FMM is poised to play a leading role in exascale computing, and we end the paper with a discussion of the features that make it a particularly favorable algorithm for the emerging heterogeneous and massively parallel architectural landscape.