Skip to content

Commit 2a2e8dd

Browse files
committed
Fix dtype everywhere Ainv is called
1 parent 092f610 commit 2a2e8dd

1 file changed

Lines changed: 15 additions & 3 deletions

File tree

simpeg/potential_fields/magnetics/simulation.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1991,13 +1991,23 @@ def _Jtvec(self, m, v, f):
19911991
if v is None:
19921992
v = np.eye(Q.shape[0])
19931993
divt_solve_q = (
1994-
self._DivT * (self._Ainv * ((Q * self.MfMuiI * -self._DivT).T * v))
1994+
self._DivT
1995+
* (
1996+
self._Ainv
1997+
* ((Q * self.MfMuiI * -self._DivT).T * v).astype(self.solver_dtype)
1998+
)
19951999
+ Q.T * v
19962000
)
19972001
del v
19982002
else:
19992003
divt_solve_q = (
2000-
self._DivT * (self._Ainv * ((-self._Div * (self.MfMuiI.T * (Q.T * v)))))
2004+
self._DivT
2005+
* (
2006+
self._Ainv
2007+
* ((-self._Div * (self.MfMuiI.T * (Q.T * v)))).astype(
2008+
self.solver_dtype
2009+
)
2010+
)
20012011
+ Q.T * v
20022012
)
20032013

@@ -2071,7 +2081,9 @@ def _Jvec(self, m, v, f):
20712081
self.MfMuiI * Mf_r_mui_deriv * v
20722082
)
20732083

2074-
Ainv_Ddm = self._Ainv * (self._Div * (-dCmu_dm + db_dm))
2084+
Ainv_Ddm = self._Ainv * (self._Div * (-dCmu_dm + db_dm)).astype(
2085+
self.solver_dtype
2086+
)
20752087

20762088
Jv = Q * (C * Ainv_Ddm + (-dCmu_dm + db_dm))
20772089

0 commit comments

Comments
 (0)