Title: A Cholesky preconditioned conjugate gradient algorithm on GPU for the 3D parabolic equation

Authors: Jiaquan Gao; Bo Li

Addresses: College of Computer Science and Technology, Zhejiang University of Technology, Hangzhou 310023, China ' College of Computer Science and Technology, Zhejiang University of Technology, Hangzhou 310023, China

Abstract: For solving the large sparse linear systems to which numerical solution of the three-dimensional (3D) parabolic equation leads, an efficient parallel preconditioned conjugate gradient algorithm with the modified incomplete Cholesky (MIC) preconditioner on the GPU (MICPCGA) is proposed. In our proposed method, for this case, we overcome the drawback that the MIC preconditioner is generally difficult to parallelise on the GPU owing to the forward/backward substitutions, and thus present an efficient parallel implementation method on the GPU (GPUFBS). In addition, the vector operations are optimised by grouping several vector operations into a single kernel, and an inner-product kernel is suggested, and a kernel for the sparse matrix-vector multiplication in the CUSPARSE library is adopted. Numerical results show that our proposed GPUFBS and MICPCGA both can achieve a significant speedup, and compared to an approximate inverse SSOR preconditioned conjugate gradient algorithm (SSORPCGA), our proposed MICPCGA not only obtains a bigger speedup, but also has higher precision in solving the 3D parabolic equation.

Keywords: conjugate gradient; graphic processing unit; GPU; 3D parabolic equations; modified incomplete Cholesky preconditioner; parallel implementation; large sparse linear systems.

DOI: 10.1504/IJCSE.2015.073493

International Journal of Computational Science and Engineering, 2015 Vol.11 No.4, pp.339 - 348

Available online: 10 Dec 2015 *

Full-text access for editors Access for subscribers Purchase this article Comment on this article