Skip to content

Commit c852fc4

Browse files
committed
feat: adapted all comm calls in Fredholm1 to new method signatures
1 parent 693f078 commit c852fc4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pylops_mpi/signalprocessing/Fredholm1.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,8 @@ def _matvec(self, x: DistributedArray) -> DistributedArray:
128128
for isl in range(self.nsls[self.rank]):
129129
y1[isl] = ncp.dot(self.G[isl], x[isl])
130130
# gather results
131-
y[:] = ncp.vstack(y._allgather(y1)).ravel()
131+
y[:] = ncp.vstack(y._allgather(y.base_comm, y.base_comm_nccl, y1,
132+
engine=y.engine)).ravel()
132133
return y
133134

134135
def _rmatvec(self, x: NDArray) -> NDArray:
@@ -165,5 +166,6 @@ def _rmatvec(self, x: NDArray) -> NDArray:
165166
y1[isl] = ncp.dot(x[isl].T.conj(), self.G[isl]).T.conj()
166167

167168
# gather results
168-
y[:] = ncp.vstack(y._allgather(y1)).ravel()
169+
y[:] = ncp.vstack(y._allgather(y.base_comm, y.base_comm_nccl, y1,
170+
engine=y.engine)).ravel()
169171
return y

0 commit comments

Comments
 (0)