warp.optim.linear.CG

class warp.optim.linear.CG(

A,

b,

x,

tol=None,

atol=None,

maxiter=0,

M=None,

callback=None,

check_every=10,

use_cuda_graph=True,

)

Pre-allocated state for the Conjugate Gradient solver.

See LinearSolverState for the constructor parameters. The preconditioner M may be freely changed (or toggled between None and a valid operator) between calls as long as the matrix shape, batch count, dtype, and device remain the same.

Parameters:

• A (array | BsrMatrix | LinearOperator)

• b (array)

• x (array)

• tol (float | None)

• atol (float | None)

• maxiter (float | None)

• M (array | BsrMatrix | LinearOperator | None)

• callback (Callable | None)

• check_every (int)

• use_cuda_graph (bool)

__init__(

A,

b,

x,

tol=None,

atol=None,

maxiter=0,

M=None,

callback=None,

check_every=10,

use_cuda_graph=True,

)

Parameters:

• A (array | BsrMatrix | LinearOperator)

• b (array)

• x (array)

• tol (float | None)

• atol (float | None)

• maxiter (float | None)

• M (array | BsrMatrix | LinearOperator | None)

• callback (Callable | None)

• check_every (int)

• use_cuda_graph (bool)

Methods

__init__(A, b, x[, tol, atol, maxiter, M, ...])