warp.optim.linear#

Iterative linear solvers for sparse linear systems.

This module provides GPU-accelerated iterative methods for solving linear systems, including conjugate gradient (CG), biconjugate gradient stabilized (BiCGSTAB), conjugate residual (CR), and generalized minimal residual (GMRES) methods.

API#

BiCGSTAB

Pre-allocated state for the BiConjugate Gradient Stabilized solver.

CG

Pre-allocated state for the Conjugate Gradient solver.

CR

Pre-allocated state for the Conjugate Residual solver.

GMRES

Pre-allocated state for the restarted Generalized Minimum Residual solver.

LinearOperator

Linear operator to be used as left-hand-side of linear iterative solvers.

LinearSolverState

Pre-allocated state for a linear iterative solver.

aslinearoperator

Cast the dense or sparse matrix A as a LinearOperator.

bicgstab

Compute an approximate solution to a linear system using the Biconjugate Gradient Stabilized method (BiCGSTAB).

cg

Compute an approximate solution to a symmetric, positive-definite linear system using the Conjugate Gradient algorithm.

cr

Compute an approximate solution to a symmetric, positive-definite linear system using the Conjugate Residual algorithm.

gmres

Compute an approximate solution to a linear system using the restarted Generalized Minimum Residual method (GMRES[k]).

preconditioner

Construct and return a preconditioner for an input matrix.