From c82c661bfe31bce19bbe5eca248ccdf9eba3fd6d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 2 Dec 2013 00:18:41 -0600
Subject: [PATCH] Allow wait_for in multi_put
---
pyopencl/array.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/pyopencl/array.py b/pyopencl/array.py
index d7525c84..2aac16ea 100644
--- a/pyopencl/array.py
+++ b/pyopencl/array.py
@@ -1755,7 +1755,8 @@ def multi_take_put(arrays, dest_indices, src_indices, dest_shape=None,
return out
-def multi_put(arrays, dest_indices, dest_shape=None, out=None, queue=None):
+def multi_put(arrays, dest_indices, dest_shape=None, out=None, queue=None,
+ wait_for=None):
if not len(arrays):
return []
@@ -1811,7 +1812,8 @@ def multi_put(arrays, dest_indices, dest_shape=None, out=None, queue=None):
+ list(flatten(
(i.base_data, i.offset)
for i in arrays[chunk_slice]))
- + [dest_indices.size]))
+ + [dest_indices.size]),
+ **dict(wait_for=wait_for))
# FIXME should wait on incoming events
--
GitLab