From c5068444ba779eca3baa39074099dc0678f1d27d Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 17 Aug 2015 15:14:53 -0500
Subject: [PATCH] Make sure nd ranges are Py ints, for legacy PyOpenCL
---
loopy/compiled.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index d8d127c0b..079da0da2 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -716,11 +716,15 @@ def generate_invoker(kernel, cl_kernel, impl_arg_info, options):
if not lsize_expr:
lsize_expr = (1,)
+ def strify_tuple(t):
+ return "(%s,)" % (
+ ", ".join("int(%s)" % strify(t_i) for t_i in t))
+
gen("_lpy_evt = _lpy_cl.enqueue_nd_range_kernel(queue, cl_kernel, "
"%(gsize)s, %(lsize)s, wait_for=wait_for, g_times_l=True)"
% dict(
- gsize=strify(gsize_expr),
- lsize=strify(lsize_expr)))
+ gsize=strify_tuple(gsize_expr),
+ lsize=strify_tuple(lsize_expr)))
gen("")
# }}}
--
GitLab