Skip to content

Commit 76c9c34

Browse files
committed
add n_threads as a kwarg on Pardiso
1 parent 7876f77 commit 76c9c34

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pymatsolver/direct/pardiso.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,15 @@ class Pardiso(Base):
1616

1717
_transposed = False
1818

19-
def __init__(self, A, **kwargs):
19+
def __init__(self, A, n_threads=None, **kwargs):
2020
super().__init__(A, **kwargs)
2121
self.solver = MKLPardisoSolver(
2222
self.A,
2323
matrix_type=self._matrixType(),
2424
factor=False
2525
)
26+
if n_threads is not None:
27+
self.n_threads = n_threads
2628

2729
def _matrixType(self):
2830
"""

0 commit comments

Comments
 (0)