Fix cl.array.multi_put.

pyopencl/array.py

@@ -1181,8 +1181,8 @@ def multi_put(arrays, dest_indices, dest_shape=None, out=None, queue=None):
 
     def make_func_for_chunk_size(chunk_size):
         knl = elementwise.get_put_kernel(
+                context,
                 a_dtype, dest_indices.dtype, vec_count=chunk_size)
-        knl.set_block_shape(*dest_indices._block)
         return knl
 
     knl = make_func_for_chunk_size(chunk_size)

pyopencl/elementwise.py

@@ -380,7 +380,7 @@ def get_put_kernel(context, dtype, idx_dtype, vec_count=1):
             + "\n".join("dest%d[dest_idx] = src%d[i];" % (i, i)
                 for i in range(vec_count)))
 
-    return get_elwise_kernel(args, body, name="put")
+    return get_elwise_kernel(context, args, body, name="put")
 
 
 @context_dependent_memoize