From 9ea089aaf7cb6368d5f605b641e67e996f96ecad Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 19 Jan 2021 19:10:50 -0600 Subject: [PATCH] Tweak examples to emphasize importance of retaining kernels --- examples/demo.py | 3 ++- examples/demo_array.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/demo.py b/examples/demo.py index 623660fe..a4a50333 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -23,7 +23,8 @@ __kernel void sum( """).build() res_g = cl.Buffer(ctx, mf.WRITE_ONLY, a_np.nbytes) -prg.sum(queue, a_np.shape, None, a_g, b_g, res_g) +knl = prg.sum # Use this Kernel object for repeated calls +knl(queue, a_np.shape, None, a_g, b_g, res_g) res_np = np.empty_like(a_np) cl.enqueue_copy(queue, res_np, res_g) diff --git a/examples/demo_array.py b/examples/demo_array.py index 41b0f79e..74bb7cfc 100644 --- a/examples/demo_array.py +++ b/examples/demo_array.py @@ -22,6 +22,7 @@ prg = cl.Program(ctx, """ } """).build() -prg.sum(queue, a.shape, None, a_dev.data, b_dev.data, dest_dev.data) +knl = prg.sum # Use this Kernel object for repeated calls +knl(queue, a.shape, None, a_dev.data, b_dev.data, dest_dev.data) print(la.norm((dest_dev - (a_dev+b_dev)).get())) -- GitLab