Skip to content

qttools.wave_function_solver.cudss#

source module qttools.wave_function_solver.cudss

Classes

  • cuDSS — Wave function solver using cuDSS for sparse matrix solving.

source class cuDSS(explicitely_reset_operands: str = 'b', use_multithreading: bool = True)

Bases : WFSolver

Wave function solver using cuDSS for sparse matrix solving.

This solver uses the cuDSS library to solve sparse linear systems on NVIDIA GPUs.

Initializes the cuDSS wave function solver.

Parameters

  • explicitely_reset_operands : str, optional — String indicating which operands to reset explicitly. If "a" is in the string, the system matrix a will be reset before solving. If "b" is in the string, the right-hand side vector b will be reset before solving. Default is "b", meaning only the right-hand side vector will be reset.

  • use_multithreading : bool, optional — Whether to use multithreading for the solver. If True, it will attempt to find a suitable multithreading library. Default is True.

Methods

  • solve — Solves the sparse linear system a @ x = b using cuDSS.

source method cuDSS.solve(a: sparse.spmatrix, b: NDArray) → NDArray

Solves the sparse linear system a @ x = b using cuDSS.

Parameters

  • a : sparse.spmatrix — The sparse system matrix.

  • b : NDArray — The right-hand side vector.

Returns

  • x : NDArray — The solution vector.