A simple GPU implementation of spectral-element methods for solving 3D Poisson type equations on rectangular domains and its applications

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liu, Xinyu, Shen, Jie, Zhang, Xiangxiong
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913387153195008
author Liu, Xinyu
Shen, Jie
Zhang, Xiangxiong
author_facet Liu, Xinyu
Shen, Jie
Zhang, Xiangxiong
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2310_00226
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle A simple GPU implementation of spectral-element methods for solving 3D Poisson type equations on rectangular domains and its applications
Liu, Xinyu
Shen, Jie
Zhang, Xiangxiong
Numerical Analysis
Distributed, Parallel, and Cluster Computing
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.
title A simple GPU implementation of spectral-element methods for solving 3D Poisson type equations on rectangular domains and its applications
topic Numerical Analysis
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2310.00226