Parallel 3-dim fast Fourier transforms with load balancing of the plane waves

Abstract

The plane wave method is most widely used for solving the Kohn-Sham equations in first-principles materials science computations. In this procedure, the three-dimensional (3-dim) trial wave functions' fast Fourier transform (FFT) is a regular operation and one of the most demanding algorithms in terms of the scalability on a parallel machine. We propose a new partitioning algorithm for the 3-dim FFT grid to accomplish the trade-off between the communication overhead and load balancing of the plane waves. It is shown by qualitative analysis and numerical results that our approach could scale the plane wave first-principles calculations up to more nodes.

Cited by in corpus (5)

• Applicability of Kerker preconditioning scheme to the self-consistent density functional theory calculations of inhomogeneous systems
• A force-based gradient descent method for $\text{ab initio}$ atomic structure relaxation
• New zirconium hydrides predicted by structure search method based on first principles calculations
• A first principles study of high pressure phase diagram of bismuth at 0 ~14 GPa based on structure search and quasi-harmonic approximation
• Projected gradient descent algorithm for $\text{ab initio}$ crystal structure relaxation under a fixed unit cell volume