Fixed interp bug introduced in last commit.

mikgroup · Jul 29, 2018 · 79917d7 · 79917d7
1 parent 966b0df
commit 79917d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sigpy/interp.py b/sigpy/interp.py
@@ -44,7 +44,7 @@ def interp(input, width, table, coord):
         if device == util.cpu_device:
             _interp(output, input, width, table, coord)
         else:
-            _interp(output, input, width, table, coord, size=npts)
+            _interp(output, input, width, table, coord, size=npts * batch)
 
         return output.reshape(batch_shape + pts_shape)