Fix zeros() on vector array. (Reported by Sebastian Nowozin)

pyopencl/array.py

@@ -680,7 +680,8 @@ def zeros(*args, **kwargs):
     def _zeros(queue, shape, dtype, order="C", allocator=None):
         result = Array(queue, shape, dtype,
                 order=order, allocator=allocator)
-        result.fill(0)
+        zero = np.zeros((), dtype)
+        result.fill(zero)
         return result
 
     if isinstance(args[0], cl.Context):
@@ -703,7 +704,8 @@ def empty_like(ary):
 
 def zeros_like(ary):
     result = empty_like(ary)
-    result.fill(0)
+    zero = np.zeros((), ary.dtype)
+    result.fill(zero)
     return result
 
 

test/test_array.py

@@ -615,6 +615,8 @@ def test_vector_fill(ctx_getter):
     a = a_gpu.get()
     assert a.dtype is cl_array.vec.float4
 
+    a_gpu = cl_array.zeros(queue, 100, dtype=cl_array.vec.float4)
+