adding suitesparse::spqr

polyfem · Jan 4, 2025 · d96792b · d96792b
1 parent 760ff26
commit d96792b
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -74,6 +74,7 @@ option(POLYSOLVE_WITH_ACCELERATE    "Enable Apple Accelerate" ${POLYSOLVE_ON_APP
 option(POLYSOLVE_WITH_CHOLMOD       "Enable Cholmod library"                             ON)
 option(POLYSOLVE_WITH_UMFPACK       "Enable UmfPack library"                             ON)
 option(POLYSOLVE_WITH_SUPERLU       "Enable SuperLU library"                             ON)
+option(POLYSOLVE_WITH_SPQR          "Enable SPQR library"                                ON)
 option(POLYSOLVE_WITH_MKL           "Enable MKL library"  ${POLYSOLVE_NOT_ON_APPLE_SILICON})
 option(POLYSOLVE_WITH_CUSOLVER      "Enable cuSOLVER library"                           OFF)
 option(POLYSOLVE_WITH_PARDISO       "Enable Pardiso library"                            OFF)
@@ -269,6 +270,17 @@ if(POLYSOLVE_WITH_SUPERLU)
     endif()
 endif()
 
+# SuperLU solver
+if(POLYSOLVE_WITH_SPQR)
+    include(SPQR)
+    if(TARGET SuiteSparse::SPQR)
+        target_link_libraries(polysolve_linear PRIVATE SuiteSparse::SPQR)
+        target_compile_definitions(polysolve_linear PUBLIC POLYSOLVE_WITH_SPQR)
+    else()
+        message(WARNING "SPQR Not found, solver will not be available.")
+    endif()
+endif()
+
 # AMGCL solver
 if(POLYSOLVE_WITH_AMGCL)
     include(amgcl)

diff --git a/cmake/recipes/spqr.cmake b/cmake/recipes/spqr.cmake
@@ -0,0 +1,11 @@
+# SPQR solver
+
+if(TARGET SparseSuite::SPQR)
+    return()
+endif()
+
+message(STATUS "Third-party: creating targets 'SuiteSparse::SPQR'")
+
+# We do not have a build recipe for this, so find it as a system installed library.
+find_package(SPQR)
+
diff --git a/src/polysolve/linear/Solver.cpp b/src/polysolve/linear/Solver.cpp
@@ -27,6 +27,24 @@ template <typename _MatrixType, typename _OrderingType = COLAMDOrdering<typename
 #ifdef POLYSOLVE_WITH_UMFPACK
 #include <Eigen/UmfPackSupport>
 #endif
+#ifdef POLYSOLVE_WITH_SPQR
+#include <Eigen/SPQRSupport>
+namespace polysolve::linear {
+    template <>
+    void EigenDirect<Eigen::SPQR<StiffnessMatrix>>::analyze_pattern(const StiffnessMatrix& A, const int precond_num) {
+        m_Solver.compute(A);
+    }
+    template <>
+    void EigenDirect<Eigen::SPQR<StiffnessMatrix>>::factorize(const StiffnessMatrix &A)
+    {
+        m_Solver.compute(A);
+        if (m_Solver.info() == Eigen::NumericalIssue)
+        {
+            throw std::runtime_error("[EigenDirect] NumericalIssue encountered.");
+        }
+    }
+}
+#endif
 #ifdef POLYSOLVE_WITH_SUPERLU
 #include <Eigen/SuperLUSupport>
 #endif
@@ -345,6 +363,12 @@ namespace polysolve::linear
         {
             RETURN_DIRECT_SOLVER_PTR(SuperLU, "Eigen::SuperLU");
 #endif
+#ifdef POLYSOLVE_WITH_SPQR
+        }
+        else if (solver == "Eigen::SPQR")
+        {
+            RETURN_DIRECT_SOLVER_PTR(SPQR, "Eigen::SPQR");
+#endif
 #ifdef POLYSOLVE_WITH_MKL
         }
         else if (solver == "Eigen::PardisoLLT")
@@ -492,6 +516,9 @@ namespace polysolve::linear
 #ifdef POLYSOLVE_WITH_SUPERLU
             "Eigen::SuperLU",
 #endif
+#ifdef POLYSOLVE_WITH_SPQR
+            "Eigen::SPQR",
+#endif
 #ifdef POLYSOLVE_WITH_MKL
             "Eigen::PardisoLLT",
             "Eigen::PardisoLDLT",