Generic interface

Krylov.jl provides a generic interface for solving linear systems using (block) Krylov methods. The interface is designed to be common for all methods and contains three routines krylov_workspace, krylov_solve and krylov_solve!. They allow to build Krylov workspaces and call both in-place and out-of-place variants of the solvers with a unified API.

Krylov.krylov_workspace — Function

workspace = krylov_workspace(method, args...; kwargs...)
workspace = krylov_workspace(Val(method), args...; kwargs...)

Generic function that dispatches to the appropriate workspace constructor for each subtype of KrylovWorkspace and BlockKrylovWorkspace. In both calls, method is a symbol (such as :cg, :gmres or :block_minres) that specifies the (block) Krylov method for which a workspace is desired. The returned workspace can later be used by krylov_solve! to execute the (block) Krylov method in-place.

Pass the plain symbol method for convenience, or wrap it in Val(method) to enable full compile-time specialization, improve type inference, and achieve maximum performance.

source

Krylov.krylov_solve — Function

krylov_solve(method, args...; kwargs...)
krylov_solve(Val(method), args...; kwargs...)

Generic function that dispatches to the appropriate out-of-place (block) Krylov method specified by method. In both calls, method is a symbol (such as :cg, :gmres or :block_minres).

Pass the plain symbol method for convenience, or wrap it in Val(method) to enable full compile-time specialization, improve type inference, and achieve maximum performance.

source

Krylov.krylov_solve! — Function

krylov_solve!(workspace, args...; kwargs...)

Generic function that dispatches to the appropriate in-place (block) Krylov method based on the type of workspace. The argument workspace must be a subtype of KrylovWorkspace or BlockKrylovWorkspace (such as CgWorkspace, GmresWorkspace or BlockMinresWorkspace).

source

The section on workspace accessors describes how to retrieve the solution, statistics, and other results from a workspace after calling krylov_solve!.

Examples

using Krylov, SparseArrays, LinearAlgebra

# Define a symmetric positive definite matrix A and a right-hand side vector b
n = 1000
A = sprandn(n, n, 0.005)
A = A * A' + I
b = randn(n)

# Out-of-place interface
for method in (:cg, :cr, :car)
    x, stats = krylov_solve(Val(method), A, b)
    r = b - A * x
    println("Residual norm for $(method): ", norm(r))
end

Residual norm for cg: 4.6090522511157426e-7
Residual norm for cr: 4.784893762474886e-7
Residual norm for car: 3.9133091610635434e-7

using Krylov, SparseArrays, LinearAlgebra

# Define a square nonsymmetric matrix A and a right-hand side vector b
n = 100
A = sprand(n, n, 0.05) + I
b = rand(n)

# In-place interface
for method in (:bicgstab, :gmres, :qmr)
    # Create a workspace for the Krylov method
    workspace = krylov_workspace(Val(method), A, b)

    # Solve the system in-place
    krylov_solve!(workspace, A, b)

    # Get the statistics
    stats = Krylov.statistics(workspace)

    # Retrieve the solution
    x = Krylov.solution(workspace)

    # Check if the solver converged
    solved = Krylov.issolved(workspace)
    println("Convergence of $method: ", solved)

    # Display the number of iterations
    niter = Krylov.iteration_count(workspace)
    println("Number of iterations for $method: ", niter)

    # Display the elapsed timer
    timer = Krylov.elapsed_time(workspace)
    println("Elapsed time for $method: ", timer, " seconds")

    # Display the number of operator-vector products with A and A'
    nAprod = Krylov.Aprod_count(workspace)
    nAtprod = Krylov.Atprod_count(workspace)
    println("Number of operator-vector products with A and A' for $method: ", (nAprod, nAtprod))

    println()
end

Convergence of bicgstab: true
Number of iterations for bicgstab: 175
Elapsed time for bicgstab: 0.000290835 seconds
Number of operator-vector products with A and A' for bicgstab: (350, 0)

Convergence of gmres: true
Number of iterations for gmres: 82
Elapsed time for gmres: 0.000240041 seconds
Number of operator-vector products with A and A' for gmres: (82, 0)

Convergence of qmr: true
Number of iterations for qmr: 111
Elapsed time for qmr: 0.000151745 seconds
Number of operator-vector products with A and A' for qmr: (111, 111)