Acceleration of Heat Diffusion Simulation on Single-Node Systems

HPC, CUDA, OpenMP, Multi-GPU, SIMD

k-Wave is a MATLAB toolbox for simulating the propagation of acoustic waves in the time domain field. It consists of several computationally intensive simulations, one of which is a heat diffusion simulation. To reduce the overall computation time and cost, an accelerated implementation targeting single-node systems was created. It allows to exploit the full potential of the CPUs by using OpenMP utilizing SIMD units and all cores though thread-level parallelism, and FFTW-like libraries for FFT computation. Alternatively, performing the computation on single and multiple GPUs is also supported on NVIDIA platforms such as DGX-2 or HGX A100 by using CUDA and the cuFFT library. As a result, the simulation output can be obtained faster at a lower price.