Skip to content

Commit

Permalink
minor: remove asarray from MPIVStack
Browse files Browse the repository at this point in the history
  • Loading branch information
mrava87 committed Jun 2, 2024
1 parent 076868d commit 45186ef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pylops_mpi/basicoperators/VStack.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,7 +136,7 @@ def _rmatvec(self, x: DistributedArray) -> DistributedArray:
y1 = []
for iop, oper in enumerate(self.ops):
y1.append(oper.rmatvec(x.local_array[self.nnops[iop]: self.nnops[iop + 1]]))
y1 = ncp.sum(ncp.asarray(y1), axis=0)
y1 = ncp.sum(y1, axis=0)
y[:] = self.base_comm.allreduce(y1, op=MPI.SUM)
return y

Expand Down

0 comments on commit 45186ef

Please sign in to comment.