Skip to content

Commit 1ef09ab

Browse files
committed
converted Bcast into bcast
1 parent 58d3ceb commit 1ef09ab

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pylops_mpi/basicoperators/MatrixMult.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -496,6 +496,7 @@ def __init__(
496496
self._col_comm = base_comm.Split(color=self._col_id, key=self._row_id)
497497

498498
self.A = A.astype(np.dtype(dtype))
499+
if saveAt: self.At = A.T.conj()
499500

500501
self.N = self._col_comm.allreduce(A.shape[0])
501502
self.K = self._row_comm.allreduce(A.shape[1])
@@ -571,8 +572,8 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
571572
for k in range(self._P_prime):
572573
Atemp = self.A.copy() if self._col_id == k else np.empty_like(self.A)
573574
Xtemp = x_block.copy() if self._row_id == k else np.empty_like(x_block)
574-
self._row_comm.Bcast(Atemp, root=k)
575-
self._col_comm.Bcast(Xtemp, root=k)
575+
self._row_comm.bcast(Atemp, root=k)
576+
self._col_comm.bcast(Xtemp, root=k)
576577
Y_local += ncp.dot(Atemp, Xtemp)
577578

578579
Y_local_unpadded = Y_local[:local_n, :local_m]

0 commit comments

Comments
 (0)