as_strided: Deal with length-zero shape/strides

inducer · Jun 6, 2016 · 1f6d1d1 · 1f6d1d1
1 parent 26eb068
commit 1f6d1d1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyopencl/array.py b/pyopencl/array.py
@@ -1711,8 +1711,10 @@ def as_strided(ary, shape=None, strides=None):
 
     # undocumented for the moment
 
-    shape = shape or ary.shape
-    strides = strides or ary.strides
+    if shape is None:
+        shape = ary.shape
+    if strides is None:
+        strides = ary.strides
 
     return Array(ary.queue, shape, ary.dtype, allocator=ary.allocator,
             data=ary.data, strides=strides)