petsc_ksp.py¶
OpenMDAO LinearSolver that uses PetSC KSP to solve for a system’s derivatives. This solver can be used under MPI.
-
class
openmdao.solvers.petsc_ksp.
Monitor
(ksp)[source]¶ Bases:
object
Prints output from PETSc’s KSP solvers
-
class
openmdao.solvers.petsc_ksp.
PetscKSP
[source]¶ Bases:
openmdao.solvers.solver_base.LinearSolver
OpenMDAO LinearSolver that uses PetSC KSP to solve for a system’s derivatives. This solver can be used under MPI.
Options: options[‘atol’] : float(1e-12)
Absolute convergence tolerance.
options[‘iprint’] : int(0)
Set to 0 to disable printing, set to 1 to print the residual to stdout each iteration, set to 2 to print subiteration residuals as well.
options[‘maxiter’] : int(100)
Maximum number of iterations.
options[‘mode’] : str(‘auto’)
Derivative calculation mode, set to ‘fwd’ for forward mode, ‘rev’ for reverse mode, or ‘auto’ to let OpenMDAO determine the best mode.
options[‘rtol’] : float(1e-12)
Relative convergence tolerance.
-
apply
(mat, sol_vec, rhs_vec)[source]¶ Applies preconditioner
Args: sol_vec : PetSC Vector
Incoming vector
rhs_vec : PetSC Vector
Empty vector into which we return the preconditioned sol_vec
-
mult
(mat, arg, result)[source]¶ KSP Callback: applies Jacobian matrix. Mode is determined by the system.
Args: arg : PetSC Vector
Incoming vector
result : PetSC Vector
Empty array into which we place the matrix-vector product.
-
solve
(rhs_mat, system, mode)[source]¶ Solves the linear system for the problem in self.system. The full solution vector is returned.
Args: rhs_mat : dict of ndarray
Dictionary containing one ndarry per top level quantity of interest. Each array contains the right-hand side for the linear solve.
system : System
Parent System object.
mode : string
Derivative mode, can be ‘fwd’ or ‘rev’.
Returns: dict of ndarray : Solution vectors
-