diff --git a/test/test_wrapper.py b/test/test_wrapper.py index 406cd046723e02239aaadf65b199b7f4b26f3540..2e3636cb11019e4d152e42b9ba1fccae4ce77790 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