GPU acceleration of ab initio simulations of large-scale identical particles based on path integral molecular dynamics

Yunuo Xiong

2024-04-03

Abstract:Path integral Monte Carlo (PIMC) and path integral molecular dynamics (PIMD) provide the golden standard for the ab initio simulations of identical particles. In this work, we achieved significant GPU acceleration based on PIMD, which is equivalent to PIMC in the ab initio simulations, and developed an open-source PIMD code repository that does not rely on any other third party library. Numerical experiments show that for a system of 1600 interacting identical bosons in a harmonic trap, using a single GPU and a single CPU, it only takes two hours to achieve satisfactory simulation accuracy. With the increase of the number of identical particles, the advantage of GPU acceleration over CPU becomes more obvious, making it possible to simulate tens of thousands of identical particles from first principles using a single GPU. For example, for a system of 10000 non-interacting bosons, numerical experiments show that it takes 23 hours to obtain a simulation that is highly consistent with the exact results. Our study shows that GPU acceleration can lay a solid foundation for the wide application of PIMD simulations for extremely large-scale identical particle quantum systems with more than 10,000 particles. Numerical experiments show that a 24GB GPU can simulate up to 40000 identical particles from first principles, and the GPU acceleration leads to a roughly linear relationship between the computation time and the number of identical particles. In addition, we have also successfully implemented simulations for fictitious identical particle thermodynamics using GPU to overcome the Fermion sign problem, which makes it promising to efficiently and accurately simulate tens of thousands of fermions based on GPU.

Computational Physics,Quantum Gases,Chemical Physics

What problem does this paper attempt to address?

This paper primarily addresses the issue of computational efficiency in quantum system simulations, particularly for the path integral molecular dynamics (PIMD) simulation of a large number of identical particles, such as bosons. Traditional PIMD or path integral Monte Carlo (PIMC) methods require massive CPU parallel computing when dealing with large-scale quantum systems, which limits their applicability. The paper proposes an acceleration method based on Graphics Processing Units (GPUs), which achieves significant speedup for PIMD simulations. An open-source PIMD code library is also developed, independent of any third-party libraries. Through this approach, the paper demonstrates that a satisfactory simulation accuracy can be achieved within two hours for a system of 1600 interacting bosons using a single GPU and CPU. As the number of particles increases, the advantage of GPU acceleration becomes more pronounced, enabling the simulation of systems with tens of thousands of particles using a single GPU. For instance, for 10,000 non-interacting bosons, the simulated results with GPU acceleration closely match the exact results, requiring only 23 hours. Furthermore, the paper discusses how to overcome the fermionic sign problem efficiently and accurately simulate the thermodynamic properties of a large number of fermions using GPU. The study finds that GPU acceleration is not only applicable to boson systems but also to simulations of fictitious particles (including fermions), providing a technical foundation for studying the thermodynamic properties of large-scale fermionic systems. In conclusion, the paper addresses the issue of utilizing GPUs to improve computational speed and reduce computation time in large-scale quantum system simulations, making it feasible to perform ab initio simulations of quantum systems containing thousands to tens of thousands of particles.

GPU acceleration of ab initio simulations of large-scale identical particles based on path integral molecular dynamics

The Implementation of the Three-Dimensional Unified Gas-Kinetic Wave-Particle Method on Multiple Graphics Processing Units

Parallelized Implementation of the Finite Particle Method for Explicit Dynamics in GPU

Implementation of the moving particle semi-implicit method for free-surface flows on GPU clusters.

Efficient molecular dynamics simulations with many-body potentials on graphics processing units

86 PFLOPS Deep Potential Molecular Dynamics simulation of 100 million atoms with ab initio accuracy

Graphics Processing Unit Acceleration and Parallelization of GENESIS for Large-Scale Molecular Dynamics Simulations

Molecular dynamics simulation of macromolecules using graphics processing unit

Accelerating Dissipative Particle Dynamics Simulations on GPUs: Algorithms, Numerics and Applications

GPU_PBTE: an Efficient Solver for Three and Four Phonon Scattering Rates on Graphics Processing Units

Robust and effective ab initio molecular dynamics simulations on the GPU cloud infrastructure using the Schrödinger Materials Science Suite

GPGPU Supported Cooperative Acceleration in Molecular Dynamics

Ab initio path integral Monte Carlo simulations of the uniform electron gas on large length scales

Particle-resolved thermal lattice Boltzmann simulation using OpenACC on multi-GPUs

Pushing the limit of molecular dynamics with ab initio accuracy to 100 million atoms with machine learning

Multi-GPU RI-HF Energies and Analytic Gradients $-$ Towards High Throughput Ab Initio Molecular Dynamics

A GPU Poisson-Fermi Solver for Ion Channel Simulations

GPU Acceleration of Numerical Atomic Orbitals-Based Density Functional Theory Algorithms within the ABACUS package

Accelerating Relativistic Exact-Two-Component Density Functional Theory Calculations with Graphical Processing Units