Skip to content

Commit be75cc5

Browse files
authored
Merge pull request #657 from mrava87/bug-conv1d
bug: fix pad in convolve1d to work with cupy
2 parents 21b3930 + 30fb3eb commit be75cc5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pylops/signalprocessing/convolve1d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,11 +144,11 @@ def __init__(
144144
self.offset -= 1
145145
self.hstar = ncp.flip(self.h, axis=-1)
146146

147-
self.pad = ncp.zeros((len(dims), 2), dtype=int)
147+
self.pad = np.zeros((len(dims), 2), dtype=int)
148148
self.pad[self.axis, 0] = max(self.offset, 0)
149149
self.pad[self.axis, 1] = -min(self.offset, 0)
150150

151-
self.padd = ncp.zeros((len(dims), 2), dtype=int)
151+
self.padd = np.zeros((len(dims), 2), dtype=int)
152152
self.padd[self.axis, 1] = max(self.offset, 0)
153153
self.padd[self.axis, 0] = -min(self.offset, 0)
154154

0 commit comments

Comments
 (0)