add n_threads as a kwarg on Pardiso

simpeg · Sep 28, 2024 · 76c9c34 · 76c9c34
1 parent 7876f77
commit 76c9c34
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymatsolver/direct/pardiso.py b/pymatsolver/direct/pardiso.py
@@ -16,13 +16,15 @@ class Pardiso(Base):
 
     _transposed = False
 
-    def __init__(self, A, **kwargs):
+    def __init__(self, A, n_threads=None, **kwargs):
         super().__init__(A, **kwargs)
         self.solver = MKLPardisoSolver(
             self.A,
             matrix_type=self._matrixType(),
             factor=False
         )
+        if n_threads is not None:
+            self.n_threads = n_threads
 
     def _matrixType(self):
         """