DOE PAGES title logo U.S. Department of Energy
Office of Scientific and Technical Information

Title: Multistage mixed precision iterative refinement

Journal Article · · Numerical Linear Algebra with Applications
DOI: https://doi.org/10.1002/nla.2434 · OSTI ID:1996581

Abstract Low precision arithmetic, in particular half precision (16‐bit) floating point arithmetic, is now available in commercial hardware. Using lower precision can offer significant savings in computation and communication costs with proportional savings in energy. Motivated by this, there has been a renewed interest in mixed precision iterative refinement schemes for solving linear systems , and new variants of GMRES‐based iterative refinement have been developed. Each particular variant with a given combination of precisions leads to different condition number‐based constraints for convergence of the backward and forward errors, and each has different performance costs. The constraints for convergence given in the literature are, as an artifact of the analyses, often overly strict in practice, and thus could lead a user to select a more expensive variant when a less expensive one would have sufficed. In this work, we develop a multistage mixed precision iterative refinement solver which aims to combine existing mixed precision approaches to balance performance and accuracy and improve usability. For a user‐specified initial combination of precisions, the algorithm begins with the least expensive approach and convergence is monitored via inexpensive computations with quantities produced during the iteration. If slow convergence or divergence is detected using particular stopping criteria, the algorithm switches to use a more expensive, but more reliable variant. A novel aspect of our approach is that, unlike existing implementations, our algorithm first attempts to use “stronger” GMRES‐based solvers for the solution update before resorting to increasing the precision(s). In some scenarios, this can avoid the need to refactorize the matrix in higher precision. We perform extensive numerical experiments on a variety of random dense problems and problems from real applications which confirm the benefits of the multistage approach.

Sponsoring Organization:
USDOE
OSTI ID:
1996581
Journal Information:
Numerical Linear Algebra with Applications, Journal Name: Numerical Linear Algebra with Applications Journal Issue: 4 Vol. 29; ISSN 1070-5325
Publisher:
Wiley Blackwell (John Wiley & Sons)Copyright Statement
Country of Publication:
United Kingdom
Language:
English

References (17)

Iterative refinement implies numerical stability journal September 1977
Solution of real and complex systems of linear equations journal May 1966
Towards dense linear algebra for hybrid GPU accelerated manycore systems journal June 2010
Prompt Report on Exa-Scale HPL-AI Benchmark conference September 2020
Exploiting the Performance of 32 bit Floating Point Arithmetic in Obtaining 64 bit Accuracy (Revisiting Iterative Refinement for Linear Systems) conference November 2006
Harnessing GPU Tensor Cores for Fast FP16 Arithmetic to Speed up Mixed-Precision Iterative Refinement Solvers conference November 2018
Error Analysis of MLFMA With Closed-Form Expressions journal October 2021
Modified Gram-Schmidt (MGS), Least Squares, and Backward Stability of MGS-GMRES journal January 2006
A New Analysis of Iterative Refinement and Its Application to Accurate Solution of Ill-Conditioned Sparse Linear Systems journal January 2017
Accelerating the Solution of Linear Systems by Iterative Refinement in Three Precisions journal January 2018
Squeezing a Matrix into Half Precision, with an Application to Solving Linear Systems journal January 2019
Simulating Low Precision Floating-Point Arithmetic journal January 2019
Three-Precision GMRES-Based Iterative Refinement for Least Squares Problems journal January 2020
Error bounds from extra-precise iterative refinement journal June 2006
Extra-Precise Iterative Refinement for Overdetermined Least Squares Problems journal February 2009
Algorithm 915, SuiteSparseQR journal November 2011
Iterative Refinement in Floating Point journal April 1967

Similar Records

Related Subjects