arrow
Volume 36, Issue 5
A Simple GPU Implementation of Spectral-Element Methods for Solving 3D Poisson Type Equations on Rectangular Domains and Its Applications

Xinyu Liu, Jie Shen & Xiangxiong Zhang

Commun. Comput. Phys., 36 (2024), pp. 1157-1185.

Published online: 2024-12

Export citation
  • Abstract

It is well known since 1960s that by exploring the tensor product structure of the discrete Laplacian on Cartesian meshes, one can develop a simple direct Poisson solver with an $\mathcal{O}(N^{\frac{d+1}{d}})$ complexity in $d$-dimension, where $N$ is the number of the total unknowns. The GPU acceleration of numerically solving PDEs has been explored successfully around fifteen years ago and become more and more popular in the past decade, driven by significant advancement in both hardware and software technologies, especially in the recent few years. We present in this paper a simple but extremely fast MATLAB implementation on a modern GPU, which can be easily reproduced, for solving 3D Poisson type equations using a spectral-element method. In particular, it costs less than one second on a Nvidia A100 for solving a Poisson equation with one billion degree of freedoms. We also present applications of this fast solver to solve a linear (time-independent) Schrödinger equation and a nonlinear (time-dependent) Cahn-Hilliard equation.

  • AMS Subject Headings

65F05, 65F60, 65Y05, 65M06, 65M60, 65M70, 65N06, 65N30, 65N35

  • Copyright

COPYRIGHT: © Global Science Press

  • Email address
  • BibTex
  • RIS
  • TXT
@Article{CiCP-36-1157, author = {Liu , XinyuShen , Jie and Zhang , Xiangxiong}, title = {A Simple GPU Implementation of Spectral-Element Methods for Solving 3D Poisson Type Equations on Rectangular Domains and Its Applications}, journal = {Communications in Computational Physics}, year = {2024}, volume = {36}, number = {5}, pages = {1157--1185}, abstract = {

It is well known since 1960s that by exploring the tensor product structure of the discrete Laplacian on Cartesian meshes, one can develop a simple direct Poisson solver with an $\mathcal{O}(N^{\frac{d+1}{d}})$ complexity in $d$-dimension, where $N$ is the number of the total unknowns. The GPU acceleration of numerically solving PDEs has been explored successfully around fifteen years ago and become more and more popular in the past decade, driven by significant advancement in both hardware and software technologies, especially in the recent few years. We present in this paper a simple but extremely fast MATLAB implementation on a modern GPU, which can be easily reproduced, for solving 3D Poisson type equations using a spectral-element method. In particular, it costs less than one second on a Nvidia A100 for solving a Poisson equation with one billion degree of freedoms. We also present applications of this fast solver to solve a linear (time-independent) Schrödinger equation and a nonlinear (time-dependent) Cahn-Hilliard equation.

}, issn = {1991-7120}, doi = {https://doi.org/10.4208/cicp.OA-2024-0072}, url = {http://global-sci.org/intro/article_detail/cicp/23606.html} }
TY - JOUR T1 - A Simple GPU Implementation of Spectral-Element Methods for Solving 3D Poisson Type Equations on Rectangular Domains and Its Applications AU - Liu , Xinyu AU - Shen , Jie AU - Zhang , Xiangxiong JO - Communications in Computational Physics VL - 5 SP - 1157 EP - 1185 PY - 2024 DA - 2024/12 SN - 36 DO - http://doi.org/10.4208/cicp.OA-2024-0072 UR - https://global-sci.org/intro/article_detail/cicp/23606.html KW - 3D Poisson equation, direct solver, spectral element methods, rectangular domain, GPU, tensor matrix multiplication. AB -

It is well known since 1960s that by exploring the tensor product structure of the discrete Laplacian on Cartesian meshes, one can develop a simple direct Poisson solver with an $\mathcal{O}(N^{\frac{d+1}{d}})$ complexity in $d$-dimension, where $N$ is the number of the total unknowns. The GPU acceleration of numerically solving PDEs has been explored successfully around fifteen years ago and become more and more popular in the past decade, driven by significant advancement in both hardware and software technologies, especially in the recent few years. We present in this paper a simple but extremely fast MATLAB implementation on a modern GPU, which can be easily reproduced, for solving 3D Poisson type equations using a spectral-element method. In particular, it costs less than one second on a Nvidia A100 for solving a Poisson equation with one billion degree of freedoms. We also present applications of this fast solver to solve a linear (time-independent) Schrödinger equation and a nonlinear (time-dependent) Cahn-Hilliard equation.

Liu , XinyuShen , Jie and Zhang , Xiangxiong. (2024). A Simple GPU Implementation of Spectral-Element Methods for Solving 3D Poisson Type Equations on Rectangular Domains and Its Applications. Communications in Computational Physics. 36 (5). 1157-1185. doi:10.4208/cicp.OA-2024-0072
Copy to clipboard
The citation has been copied to your clipboard