Iterative methods in GPU-resident linear solvers for nonlinear constrained optimization

Kasia Świrydowicz, Nicholson Koukpaizan, Maksudul Alam, Shaked Regev, Michael Saunders, Slaven Peleš

Research output: Contribution to journalArticlepeer-review

Abstract

Linear solvers are major computational bottlenecks in a wide range of decision support and optimization computations. The challenges become even more pronounced on heterogeneous hardware, where traditional sparse numerical linear algebra methods are often inefficient. For example, methods for solving ill-conditioned linear systems have relied on conditional branching, which degrades performance on hardware accelerators such as graphical processing units (GPUs). To improve the efficiency of solving ill-conditioned systems, our computational strategy separates computations that are efficient on GPUs from those that need to run on traditional central processing units (CPUs). Our strategy maximizes the reuse of expensive CPU computations. Iterative methods, which thus far have not been broadly used for ill-conditioned linear systems, play an important role in our approach. In particular, we extend ideas from Arioli et al., (2007) to implement iterative refinement using inexact LU factors and flexible generalized minimal residual (FGMRES), with the aim of efficient performance on GPUs. We focus on solutions that are effective within broader application contexts, and discuss how early performance tests could be improved to be more predictive of the performance in a realistic environment.

Original languageEnglish
Article number103123
JournalParallel Computing
Volume123
DOIs
StatePublished - Mar 2025

Funding

This research has been supported by UT-Battelle, LLC, and used resources of the Oak Ridge Leadership Computing Facility under contract DE-AC05-00OR22725 with the U.S. Department of Energy (DOE), and by the Exascale Computing Project (17-SC-20-SC), a collaborative effort of the DOE Office of Science and the National Nuclear Security Administration. The authors thank Cosmin Petra and Nai-Yuan Chiang of Lawrence Livermore National Laboratory for their guidance when using the HiOp optimization solver, and Shri Abhyankar of Pacific Northwest National Laboratory for his help with using ExaGOTM software. The authors would also like to thank LungSheng Chien and Doris Pan of NVIDIA for their help with using the undocumented cusolverGLU module in the cuSOLVER library. Warm thanks also go to Phil Roth of Oak Ridge National Laboratory and Christopher Oehmen of Pacific Northwest National Laboratory for their support of this work. This research has been supported by UT-Battelle , LLC , and used resources of the Oak Ridge Leadership Computing Facility under contract DE-AC05-00OR22725 with the U.S. Department of Energy (DOE), and by the Exascale Computing Project (17-SC-20-SC), a collaborative effort of the DOE Office of Science and the National Nuclear Security Administration.

Keywords

  • ACOPF
  • Economic dispatch
  • GPU
  • Linear solver
  • Optimization

Fingerprint

Dive into the research topics of 'Iterative methods in GPU-resident linear solvers for nonlinear constrained optimization'. Together they form a unique fingerprint.

Cite this