removed now useless bcast and fixed mask in test

Author: astroC86

pylops_mpi/basicoperators/MatrixMult.py

@@ -182,9 +182,7 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
 
         my_own_cols = self._rank_col_lens[self.rank]
         x_arr = x.local_array.reshape((self.dims[0], my_own_cols))
-        x_arr = x_arr.astype(self.dtype)
-
-        X_local = self._layer_comm.bcast(x_arr if self._group_id == self._layer_id else None, root=self._layer_id)
+        X_local = x_arr.astype(self.dtype)
         Y_local = ncp.vstack(
             self._layer_comm.allgather(
                 ncp.matmul(self.A, X_local)

tests/test_matrixmult.py

@@ -80,7 +80,7 @@ def test_SUMMAMatrixMult(M, K, N, dtype_str):
         local_shapes=[K * cl_b for cl_b in all_local_col_len],
         partition=Partition.SCATTER,
         base_comm=comm,
-        mask=[i // p_prime for i in range(size)],
+        mask=[i % p_prime for i in range(size)],
         dtype=dtype
     )