mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-02 04:59:35 +08:00
rowanc1
29 lines
584 B
Python
29 lines
584 B
Python
from mumps import DMumpsContext
|
|
|
|
class MumpsSolver():
|
|
A = None
|
|
ctx = None
|
|
x = None
|
|
|
|
def __init__(self, A, **kwagrs):
|
|
|
|
self.ctx = DMumpsContext(sym=0, par=1)
|
|
|
|
if self.ctx.myid ==0:
|
|
self.A = A
|
|
self.ctx.set_icntl(14, 60)
|
|
self.ctx.set_centralized_sparse(A)
|
|
|
|
self.ctx.set_silent()
|
|
self.ctx.run(job=4) # Factorization
|
|
|
|
def solve(self,b):
|
|
self.x = b.copy()
|
|
self.ctx.set_rhs(self.x)
|
|
self.ctx.run(job=3) # Solve
|
|
|
|
return self.x
|
|
|
|
def clean(self):
|
|
self.ctx.destroy()
|