#2217 add rest of iterative solvers

pybamm-team · Sep 12, 2022 · 2c1163b · 2c1163b
1 parent 7134ecb
commit 2c1163b
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 4 deletions.
diff --git a/pybamm/solvers/c_solvers/idaklu/casadi_solver.cpp b/pybamm/solvers/c_solvers/idaklu/casadi_solver.cpp
@@ -173,6 +173,38 @@ CasadiSolver::CasadiSolver(np_array atol_np, double rel_tol,
     LS = SUNLinSol_SPBCGS(yy, precon_type, options.linsol_max_iterations);
 #endif
   }
+  else if (options.linear_solver == "SUNLinSol_SPFGMR")
+  {
+    DEBUG("\tsetting SUNLinSol_SPFGMR_linear solver");
+#if SUNDIALS_VERSION_MAJOR >= 6
+    LS = SUNLinSol_SPFGMR(yy, precon_type, options.linsol_max_iterations,
+                          sunctx);
+#else
+    LS = SUNLinSol_SPFGMR(yy, precon_type, options.linsol_max_iterations);
+#endif
+  }
+  else if (options.linear_solver == "SUNLinSol_SPGMR")
+  {
+    DEBUG("\tsetting SUNLinSol_SPGMR solver");
+#if SUNDIALS_VERSION_MAJOR >= 6
+    LS = SUNLinSol_SPGMR(yy, precon_type, options.linsol_max_iterations,
+                          sunctx);
+#else
+    LS = SUNLinSol_SPGMR(yy, precon_type, options.linsol_max_iterations);
+#endif
+  }
+  else if (options.linear_solver == "SUNLinSol_SPTFQMR")
+  {
+    DEBUG("\tsetting SUNLinSol_SPGMR solver");
+#if SUNDIALS_VERSION_MAJOR >= 6
+    LS = SUNLinSol_SPTFQMR(yy, precon_type, options.linsol_max_iterations,
+                          sunctx);
+#else
+    LS = SUNLinSol_SPTFQMR(yy, precon_type, options.linsol_max_iterations);
+#endif
+  }
+
+
 
   IDASetLinearSolver(ida_mem, LS, J);
 

diff --git a/pybamm/solvers/c_solvers/idaklu/common.hpp b/pybamm/solvers/c_solvers/idaklu/common.hpp
@@ -18,6 +18,10 @@
 #include <sunlinsol/sunlinsol_dense.h> /* access to dense linear solver          */
 #include <sunlinsol/sunlinsol_lapackdense.h> /* access to lapack linear solver          */
 #include <sunlinsol/sunlinsol_spbcgs.h> /* access to spbcgs iterative linear solver          */
+#include <sunlinsol/sunlinsol_spfgmr.h>
+#include <sunlinsol/sunlinsol_spgmr.h>
+#include <sunlinsol/sunlinsol_sptfqmr.h>
+
 #include <sunmatrix/sunmatrix_sparse.h> /* access to sparse SUNMatrix           */
 #include <sunmatrix/sunmatrix_dense.h> /* access to dense SUNMatrix           */
 

diff --git a/pybamm/solvers/c_solvers/idaklu/options.cpp b/pybamm/solvers/c_solvers/idaklu/options.cpp
@@ -35,7 +35,10 @@ Options::Options(py::dict options)
   else if (linear_solver == "SUNLinSol_KLU" && jacobian == "sparse")
   {
   }
-  else if (linear_solver == "SUNLinSol_SPBCGS" &&
+  else if ((linear_solver == "SUNLinSol_SPBCGS" ||
+            linear_solver == "SUNLinSol_SPFGMR" ||
+            linear_solver == "SUNLinSol_SPGMR" ||
+            linear_solver == "SUNLinSol_SPTFQMR") &&
            (jacobian == "sparse" || jacobian == "matrix-free"))
   {
     using_iterative_solver = true;

diff --git a/pybamm/solvers/idaklu_solver.py b/pybamm/solvers/idaklu_solver.py
@@ -52,7 +52,8 @@ class IDAKLUSolver(pybamm.BaseSolver):
             linear_solver: "SUNLinSol_KLU", # name of sundials linear solver to use
                                   # options are: "SUNLinSol_KLU",
                                   # "SUNLinSol_Dense", "SUNLinSol_LapackDense"
-                                  # "SUNLinSol_SPBCGS"
+                                  # "SUNLinSol_SPBCGS", "SUNLinSol_SPFGMR",
+                                  # "SUNLinSol_SPGMR", "SUNLinSol_SPTFQMR",
             preconditioner: "BBDP", # preconditioner for iterative solvers,
                                     # can be "none", "BBDP"
             linsol_max_iterations: 5, # for iterative linear solvers, max number of

diff --git a/tests/unit/test_solvers/test_idaklu_solver.py b/tests/unit/test_solvers/test_idaklu_solver.py
@@ -416,7 +416,8 @@ def test_options(self):
         for jacobian in ["none", "dense", "sparse", "matrix-free", "garbage"]:
                 for linear_solver in [
                         "SUNLinSol_SPBCGS", "SUNLinSol_Dense", "SUNLinSol_LapackDense",
-                        "SUNLinSol_KLU", "garbage"
+                        "SUNLinSol_KLU", "SUNLinSol_SPFGMR", "SUNLinSol_SPGMR",
+                        "SUNLinSol_SPTFQMR", "garbage"
                 ]:
                     for precon in ["none", "BBDP"]:
                         options = {
@@ -426,7 +427,7 @@ def test_options(self):
                         }
                         solver = pybamm.IDAKLUSolver(options=options)
                         soln = solver.solve(model, t_eval)
-                        np.testing.assert_array_almost_equal(soln.y, soln_base.y)
+                        np.testing.assert_array_almost_equal(soln.y, soln_base.y, 5)