From e93b1f6077080ecd07f4cfdd4a78017df9e2d14e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 30 Dec 2012 18:35:17 +0100
Subject: [PATCH] Fix cl.array.multi_put.
---
pyopencl/array.py | 2 +-
pyopencl/elementwise.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/pyopencl/array.py b/pyopencl/array.py
index 33fa2fc0..c6d19cc0 100644
--- a/pyopencl/array.py
+++ b/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)
diff --git a/pyopencl/elementwise.py b/pyopencl/elementwise.py
index 60c4bc11..77b9626b 100644
--- a/pyopencl/elementwise.py
+++ b/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
--
GitLab