autopdex.solver.linear_solve_pardiso

autopdex.solver.linear_solve_pardiso(mat, rhs, solver, verbose, free_dofs)[source]

Solve a linear system using the PARDISO solver (requires Intel MKL and pypardiso(‘lu’) or sparse_dot_mkl(‘qr’) to be installed).

This function solves a linear system using PARDISO, with options for different solver types. The input matrix is first converted to a SciPy CSR matrix, and rows and columns corresponding to Dirichlet DOFs are optionally removed.

Parameters:

  • mat (jax.experimental.sparse.BCOO) – The input matrix in JAX BCOO format.

  • rhs (jnp.ndarray) – The (reduced/free) right-hand side vector.

  • solver (str) – The type of solver to use (‘lu’ or ‘qr’).

  • verbose (int) – Verbosity level. If >= 2, timing information is printed.

  • free_dofs (array or None) – Boolean array indicating which degrees of freedom are free.

Returns:

The solution vector.

Return type:

jax.numpy.array