From e2250b8d099fa8b5eae3f8ae98017c79b7a566e2 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Fri, 31 Mar 2017 15:42:02 -0500
Subject: [PATCH] ListOfListsBuilder: Ensure count and write kernels get passed
the correct local work size of 1 when do_not_vectorize() is true.
Closes #175
---
pyopencl/algorithm.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pyopencl/algorithm.py b/pyopencl/algorithm.py
index 328a1f77..e256b578 100644
--- a/pyopencl/algorithm.py
+++ b/pyopencl/algorithm.py
@@ -1039,7 +1039,7 @@ class ListOfListsBuilder:
if self.debug:
gsize = (1,)
lsize = (1,)
- elif self.complex_kernel and queue.device.type == cl.device_type.CPU:
+ elif self.do_not_vectorize():
gsize = (4*queue.device.max_compute_units,)
lsize = (1,)
else:
--
GitLab