slice size should round up, not down

inducer · Jul 5, 2015 · acb59dc · acb59dc
1 parent 8e4f7d8
commit acb59dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycuda/gpuarray.py b/pycuda/gpuarray.py
@@ -786,7 +786,7 @@ def __getitem__(self, index):
 
                 array_stride = self.strides[array_axis]
 
-                new_shape.append((stop-start)//idx_stride)
+                new_shape.append((stop-start-1)//idx_stride+1)
                 new_strides.append(idx_stride*array_stride)
                 new_offset += array_stride*start