fix slepc v3.22+ shell matrix solving

jcmgray · Oct 28, 2024 · 20fe4c6 · 20fe4c6
1 parent 3695f50
commit 20fe4c6
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/quimb/linalg/slepc_linalg.py b/quimb/linalg/slepc_linalg.py
@@ -85,11 +85,15 @@ def __init__(self, lo):
         self.lo = lo
         self.real = lo.dtype in (float, np.float64)
 
-    def mult(self, _, x, y):
-        y[:] = self.lo.matvec(x)
-
-    def multHermitian(self, _, x, y):
-        y[:] = self.lo.rmatvec(x)
+    def mult(self, _, xvec, yvec):
+        x = xvec.getArray(readonly=True)
+        y = self.lo.matvec(x)
+        yvec.setArray(y)
+
+    def multHermitian(self, _, xvec, yvec):
+        x = xvec.getArray(readonly=True)
+        y = self.lo.rmatvec(x)
+        yvec.setArray(y)
 
 
 def linear_operator_2_petsc_shell(lo, comm=None):