Whole Program Generation of Massively Parallel Shallow Water Equation Solvers

Kuckuk S, Köstler H (2018)

Publication Language: English

Publication Type: Conference contribution

Publication year: 2018

Pages Range: 78-87

Conference Proceedings Title: 2018 IEEE International Conference on Cluster Computing (CLUSTER)

Event location: Belfast GB

ISBN: 2168-9253

URI: https://ieeexplore.ieee.org/document/8514861

DOI: 10.1109/CLUSTER.2018.00020

Abstract

The study of ocean currents has been an active area of research for decades. As a model close to the water surface, the shallow water equations (SWE) can be used. For realistic simulations, efficient numerical solvers are necessary that exhibit a good node-level performance while still maintaining scalability. When comparing the discretized model and the actual implementation, one often finds that they differ vastly. This gap makes it hard for domain experts to implement their models and high performance computing (HPC) experts are required to ensure an optimal implementation. Using domain-specific languages (DSLs) and code generation techniques can be a useful tool to bridge this gap. In recent years, ExaStencils and its DSL ExaSlang have proven to provide a suitable platform for this. We present an extension from up to now elliptic to hyperbolic partial differential equations (PDEs) in this work, namely the SWE. After setting up a suitable discretization, we demonstrate how it can be mapped to ExaSlang code. This code is still quite similar to the original, mathematically motivated specification and can be easily written by domain experts. Still, solvers generated from this abstract representation can be run on large-scale clusters. We demonstrate this by giving performance and scalability results on the state-of-the-art GPU cluster Piz Daint where we solve for close to a trillion unknowns on 2048 GPUs. From there, we discuss the performance impact of different optimizations such as overlapping computation and communication, or switching to a hybrid CPU-GPU parallelization scheme.

Authors with CRIS profile

Sebastian Kuckuk Lehrstuhl für Informatik 10 (Systemsimulation) (LSS) Harald Köstler Lehrstuhl für Informatik 10 (Systemsimulation) (LSS)

Related research project(s)

Rechenleistungsoptimierte Software-Strategien für auf unstrukturierten Gittern basierende Anwendungen in der Ozeanmodellierung (OCEAN) Jan. 1, 2017 - Sept. 30, 2020

How to cite

APA:

Kuckuk, S., & Köstler, H. (2018). Whole Program Generation of Massively Parallel Shallow Water Equation Solvers. In 2018 IEEE International Conference on Cluster Computing (CLUSTER) (pp. 78-87). Belfast, GB.

MLA:

Kuckuk, Sebastian, and Harald Köstler. "Whole Program Generation of Massively Parallel Shallow Water Equation Solvers." Proceedings of the 2018 IEEE International Conference on Cluster Computing (CLUSTER), Belfast 2018. 78-87.

BibTeX: Download