From d1d4f1eb0ec68d8574dc71d82e205ac3f24ddd8c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 29 Jun 2014 13:10:46 -0600
Subject: [PATCH] Minor fixes to test_enqueue_task
---
test/test_wrapper.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/test/test_wrapper.py b/test/test_wrapper.py
index 406cd046..2e3636cb 100644
--- a/test/test_wrapper.py
+++ b/test/test_wrapper.py
@@ -659,7 +659,7 @@ def test_enqueue_task(ctx_factory):
""").build()
knl = prg.reverse
- n = 1000000
+ n = 100
a = np.random.rand(n).astype(np.float32)
b = np.empty_like(a)
@@ -669,9 +669,10 @@ def test_enqueue_task(ctx_factory):
knl.set_args(buf1, buf2, np.int32(n))
cl.enqueue_task(queue, knl)
- cl.enqueue_read_buffer(queue, buf2, b).wait()
+ cl.enqueue_copy(queue, b, buf2).wait()
assert la.norm(a[::-1] - b) == 0
+
if __name__ == "__main__":
# make sure that import failures get reported, instead of skipping the tests.
import pyopencl # noqa
--
GitLab