AmgT

This paper proposes AmgT, a new AMG solver that utilizes the tensor core and mixed precision ability of the latest GPUs during multiple phases of the AMG algorithm. Considering that the sparse general matrix-matrix multiplication (SpGEMM) and sparse matrix-vector multiplication (SpMV) are extensively used in the setup and solve phases, respectively, we propose a novel method based on a unified sparse storage format that leverages tensor cores and their variable precision. Our method improves both the performance of GPU kernels, and also minimizes the cost of format conversion in the whole data flow of AMG. To better utilize algorithm components in existing libraries, the data format and compute kernels of the AmgT solver are incorporated into the HYPRE library.