Merge pull request #107 from PyLops/patch-cgls

bug: fix issue in CGLS when using cupy arrays (needs to be forced to …
PyLops · Oct 2, 2024 · d8e419e · d8e419e
2 parents d05ebbb + aa71545
commit d8e419e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pylops_mpi/optimization/cls_basic.py b/pylops_mpi/optimization/cls_basic.py
@@ -373,7 +373,7 @@ def step(self, x: Union[DistributedArray, StackedDistributedArray],
 
         """
 
-        a = float(self.kold / (self.q.dot(self.q.conj()) + self.damp * self.c.dot(self.c.conj())))
+        a = float(np.abs(self.kold / (self.q.dot(self.q.conj()) + self.damp * self.c.dot(self.c.conj()))))
         x += a * self.c
         self.s -= a * self.q
         damped_x = self.damp * x