We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 076868d commit 45186efCopy full SHA for 45186ef
pylops_mpi/basicoperators/VStack.py
@@ -136,7 +136,7 @@ def _rmatvec(self, x: DistributedArray) -> DistributedArray:
136
y1 = []
137
for iop, oper in enumerate(self.ops):
138
y1.append(oper.rmatvec(x.local_array[self.nnops[iop]: self.nnops[iop + 1]]))
139
- y1 = ncp.sum(ncp.asarray(y1), axis=0)
+ y1 = ncp.sum(y1, axis=0)
140
y[:] = self.base_comm.allreduce(y1, op=MPI.SUM)
141
return y
142
0 commit comments