SolverEngine¶
- class psi4.driver.p4util.SolverEngine[source]¶
Bases:
ABC
Abstract Base Class defining the API for a matrix-vector product object required by solvers.
Engines implement the correct product functions for iterative solvers that do not require the target matrix be stored directly. Classes intended to be used as an engine for
davidson_solver()
orhamiltonian_solver()
should inherit from this base class to ensure that the required methods are defined.Note
The vector referred to here is intentionally vague, the solver does not care what it is and only holds individual or sets of them. In fact an individual vector could be split across two elements in a list, such as for different spin. Whatever data type is used and individual vector should be a single element in a list such that len(list) returns the number of vector-like objects.
Methods Summary
Compute a Matrix * trial vector products
Return a new vector object.
precondition
(R_k, w_k)Apply the preconditioner to a Residual vector
residue
(X, so_prop_ints)Compute residue
vector_axpy
(X, Y)Compute scaled vector addition operation a*X + Y
Make a copy of a vector
vector_dot
(X, Y)Compute a dot product between two vectors
vector_scale
(X)Scale a vector by some factor
Methods Documentation
- abstract compute_products(X)[source]¶
Compute a Matrix * trial vector products
- Parameters:
X (List[vector]) – Trial vectors.
- Returns:
Expected by
davidson_solver()
AX (List[vector]) – The product \(A x X_{i}\) for each X_{i} in X, in that order. Where A is the hermitian matrix to be diagonalized. len(AX) == len(X)
n (int) – The number of products that were evaluated. If the object implements product caching this may be less than len(X)
Expected by
hamiltonian_solver()
H1X (List[vector]) – The product \(H1 x X_{i}\) for each X_{i} in X, in that order. Where H1 is described in
hamiltonian_solver()
.len(H1X) == len(X)
H2X (List[vector]) – The product \(H2 x X_{i}\) for each X_{i} in X, in that order. Where H2 is described in
hamiltonian_solver()
.len(H2X) == len(X)
- abstract new_vector()[source]¶
Return a new vector object.
The solver is oblivious to the data structure used for a vector this method provides the engine with a means to create vector like quantities.
The engine calls this method with no arguments. So any defined by the engine for its own use should be optional
- Returns:
X – This should be a new vector object with the correct dimensions, assumed to be zeroed out
- Return type:
singlet vector
- abstract precondition(R_k, w_k)[source]¶
Apply the preconditioner to a Residual vector
The preconditioner is usually defined as \((w_k - D_{i})^-1\) where D is an approximation of the diagonal of the matrix that is being diagonalized.
- Parameters:
R_k (single vector) – The residual vector
w_k (float) – The eigenvalue associated with this vector
- Returns:
new_X_k – The preconditioned residual vector, a correction vector that will be used to augment the guess space
- Return type:
single vector
- abstract residue(X, so_prop_ints)[source]¶
Compute residue
- Parameters:
X – The single vector to use to compute the property.
so_prop_ints – Property integrals in SO basis for the desired transition property.
prefactor – Optional float scaling factor.
- Returns:
residue – The transition property.
- Return type:
Any
- abstract vector_axpy(X, Y)[source]¶
Compute scaled vector addition operation a*X + Y
- Parameters:
a (
float
) – The scale factor applied to XX (singlet vector) – The vector which will be scaled and added to Y
Y (single vector) – The vector which the result of a*X is added to
- Returns:
Y – The solver assumes that Y is updated, and returned. So it is safe to avoid a copy of Y if possible
- Return type:
single vector
- abstract vector_copy()[source]¶
Make a copy of a vector
- Parameters:
X (single vector) – The vector to copy
- Returns:
X’ – A copy of X should be distinct object that can be modified independently of the passed object, Has the same data when returned.
- Return type:
single vector