Skip to content

Commit 45186ef

Browse files
committed
minor: remove asarray from MPIVStack
1 parent 076868d commit 45186ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pylops_mpi/basicoperators/VStack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def _rmatvec(self, x: DistributedArray) -> DistributedArray:
136136
y1 = []
137137
for iop, oper in enumerate(self.ops):
138138
y1.append(oper.rmatvec(x.local_array[self.nnops[iop]: self.nnops[iop + 1]]))
139-
y1 = ncp.sum(ncp.asarray(y1), axis=0)
139+
y1 = ncp.sum(y1, axis=0)
140140
y[:] = self.base_comm.allreduce(y1, op=MPI.SUM)
141141
return y
142142

0 commit comments

Comments
 (0)