feat: ensure y arrays are created on same engine as x

Author: mrava87

pylops_mpi/basicoperators/MatrixMult.py

@@ -335,6 +335,7 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
             local_shapes=[(self.N * c) for c in self._rank_col_lens],
             mask=x.mask,
             partition=Partition.SCATTER,
+            engine=x.engine,
             dtype=output_dtype,
             base_comm=self.base_comm
         )
@@ -372,6 +373,7 @@ def _rmatvec(self, x: DistributedArray) -> DistributedArray:
             local_shapes=[self.K * c for c in self._rank_col_lens],
             mask=x.mask,
             partition=Partition.SCATTER,
+            engine=x.engine,
             dtype=output_dtype,
             base_comm=self.base_comm
         )
@@ -573,6 +575,7 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
                              mask=x.mask,
                              local_shapes=local_shapes,
                              partition=Partition.SCATTER,
+                             engine=x.engine,
                              dtype=output_dtype,
                              base_comm=self.base_comm)
 
@@ -638,6 +641,7 @@ def _rmatvec(self, x: DistributedArray) -> DistributedArray:
             mask=x.mask,
             local_shapes=local_shapes,
             partition=Partition.SCATTER,
+            engine=x.engine,
             dtype=output_dtype,
             base_comm=self.base_comm
         )