High-performance xPU Stencil Computations in Julia

We present an efficient approach for writing architecture-agnostic parallel high-performance stencil computations in Julia, which is instantiated in the package ParallelStencil.jl.Powerful metaprogramming, costless abstractions and multiple dispatch enable writing a single code that is suitable for both productive prototyping on a single CPU thread and high performance production runs on GPU or CPU workstations or, if used in combination with distributed parallelization packages as ImplicitGlobalGrid.jl, on supercomputers.We demonstrate performance close to the theoretical upper bound on GPUs for a 3-D heat diffusion solver, which is a massive improvement over reachable performance with CUDA.jlArray programming.

Authors

Publication Details

Journal
JuliaCon Proceedings
Published
2024-10-08
DOI
https://doi.org/10.21105/jcon.00138
Citations
5
Primary Topic
Fluid Dynamics Simulations and Interactions
Type
article
Field-Weighted Citation Impact
1.87

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

High-performance xPU Stencil Computations in Julia

Ludovic Räss, Samuel Omlin
5 citations
JuliaCon Proceedings
Fluid Dynamics Simulations and Interactions
1.87
article

High-performance xPU Stencil Computations in Julia

Ludovic Räss, Samuel Omlin
article en
5 citations

Abstract

We present an efficient approach for writing architecture-agnostic parallel high-performance stencil computations in Julia, which is instantiated in the package ParallelStencil.jl.Powerful metaprogramming, costless abstractions and multiple dispatch enable writing a single code that is suitable for both productive prototyping on a single CPU thread and high performance production runs on GPU or CPU workstations or, if used in combination with distributed parallelization packages as ImplicitGlobalGrid.jl, on supercomputers.We demonstrate performance close to the theoretical upper bound on GPUs for a 3-D heat diffusion solver, which is a massive improvement over reachable performance with CUDA.jlArray programming.

JuliaCon ProceedingsVol. 6(64)
Eidgenössische Technische Hochschule Zürich, Centro Svizzero di Calcolo Scientifico, National Supercomputing Centre Singapore
Openalex Percentile: Top 16%
Fluid Dynamics Simulations and Interactions
1.87
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.