Merge branch 'main' of https://github.com/PyLops/pylops-mpi into joss

PyLops · Oct 9, 2024 · d83f08d · d83f08d
2 parents 5ee8858 + 6d3b1e8
commit d83f08d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
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

diff --git a/pylops_mpi/waveeqprocessing/MDC.py b/pylops_mpi/waveeqprocessing/MDC.py
@@ -173,5 +173,6 @@ def MPIMDC(G, nt, nv, nfreq, dt=1., dr=1., twosided=True,
                 saveGt=saveGt, conj=conj, prescaled=prescaled,
                 base_comm=base_comm,
                 _Fredholm1=MPIFredholm1,
-                args_FFT={'engine': fftengine},
+                args_FFT={"engine": fftengine},
+                args_FFT1={"engine": fftengine},
                 args_Fredholm1={'usematmul': usematmul})