PhotoNs-GPU:A GPU accelerated cosmological simulation code

Abstract

We present a GPU-accelerated cosmological simulation code, PhotoNs-GPU, based on algorithm of Particle Mesh Fast Multipole Method (PM-FMM), and focus on the GPU utilization and optimization. A proper interpolated method for truncated gravity is introduced to speed up the special functions in kernels. We verify the GPU code in mixed precision and different levels of interpolated method on GPU. A run with single precision is roughly two times faster that double precision for current practical cosmological simulations. But it could induce a unbiased small noise in power spectrum. Comparing with the CPU version of PhotoNs and Gadget-2, the efficiency of new code is significantly improved. Activated all the optimizations on the memory access, kernel functions and concurrency management, the peak performance of our test runs achieves 48% of the theoretical speed and the average performance approaches to 35% on GPU.

Accepted for publication in RAA, 7 pages, 5 figures

References in corpus (7)

• Transients from Initial Conditions in Cosmological Simulations
• Simulating cosmic structure formation with the GADGET-4 code
• The halo mass function from the dark ages through the present day
• The Uchuu Simulations: Data Release 1 and Dark Matter Halo Concentrations
• Universal structure of dark matter haloes over a mass range of 20 orders of magnitude
• Gravitational tree-code on graphics processing units: implementation in CUDA
• A Hybrid Fast Multipole Method for Cosmological N-body Simulations

Cited by in corpus (1)

• pyC${}^2$Ray: A flexible and GPU-accelerated Radiative Transfer Framework for Simulating the Cosmic Epoch of Reionization