Solving Multiple Tridiagonal Systems on a Multi-GPU Platform
Adrián Perez Diéguez, Margarita Amor, Ramón Doallo Biempica · 2018
This work describes an efficient implementation for solving tridiagonal systems on Graphics Processing Units (GPUs). The Wang and Mou algorithm has a computation and communication pattern that matches very well to the GPU features. Thus, an implementation of this algorithm is presented here for solving large problem sizes, i.e., larger than CUDA GPU's shared memory capacity, in a Multiple-GPU environment. Also, small and medium problem sizes can take advantage of this implementation. Finally, this proposal has been tuned in order to obtain the maximum performance, resulting in a compact implementation that outperforms the CUSPARSE library (6.33x).