diff --git a/pyopencl/elementwise.py b/pyopencl/elementwise.py index 398936ff52587b91a48d0febec06a7e172d80019..05ec0f22a0ec28979a4a4e91bb5c0228729dfd00 100644 --- a/pyopencl/elementwise.py +++ b/pyopencl/elementwise.py @@ -383,7 +383,7 @@ def get_take_put_kernel(context, dtype, idx_dtype, with_offsets, vec_count=1): args = [ VectorArg(dtype, "dest%d" % i) - for i in range(vec_count) + for i in range(vec_count) ] + [ VectorArg(idx_dtype, "gmem_dest_idx", with_offset=True), VectorArg(idx_dtype, "gmem_src_idx", with_offset=True),