DOI: 10.1145/3860899 ISSN: 0098-3500
Mixed-Precision SVD on GPUs via Ogita–Aishima Iterative Refinement
Angelika Schwarz, Rasmus Munk Larsen, Samuel Rodriguez, Lev Kruglyak, Christopher BakerThe Ogita–Aishima iterative refinement algorithm for SVD is based on matrix–matrix multiplication, making it an attractive approach for GPUs. We extend the algorithm to complex arithmetic and add cluster handling. The latter is based on Wedin's perturbation theory and a per-pair version of Ogita–Aishima's convergence analysis. Earlier implementations require well-separated singular values; the proposed cluster handling removes this restriction. Our implementation refines an FP32-quality SVD computed with cuSolver's
cusolverDnXgesvdp
to FP64-quality, while matching the accuracy of the corresponding FP64 routine on all test problems—twelve synthetic singular-value distributions and 38 matrices from the SuiteSparse collection. The performance depends on the spectrum and the FP32-to-FP64 ratio. Under Ozaki-II FP64 emulation on an NVIDIA RTX PRO 6000 Blackwell workstation GPU, which has a
cusolverDnXgesvdp
on spectra whose adjacent gaps are resolved by the FP32 SVD, and at most