From 9d6dd85682cd8668781c83bf59b40c2a3a85fbf6 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 16 Jul 2015 11:57:31 -0500
Subject: [PATCH] Bitonic: shrink amount of used local memory in accordance
with @inferrna's comments in #80
---
pyopencl/bitonic_sort.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/pyopencl/bitonic_sort.py b/pyopencl/bitonic_sort.py
index 8e2b4045..d4beaba2 100644
--- a/pyopencl/bitonic_sort.py
+++ b/pyopencl/bitonic_sort.py
@@ -121,8 +121,8 @@ class BitonicSort(object):
if aux:
last_evt = knl(
queue, (nt,), wg, arr.data, idx.data,
- cl.LocalMemory(wg[0]*4*arr.dtype.itemsize),
- cl.LocalMemory(wg[0]*4*idx.dtype.itemsize),
+ cl.LocalMemory(wg[0]*arr.dtype.itemsize),
+ cl.LocalMemory(wg[0]*idx.dtype.itemsize),
wait_for=[last_evt])
for knl, nt, wg, _ in run_queue[1:]:
last_evt = knl(
@@ -184,9 +184,9 @@ class BitonicSort(object):
available_lmem = dev.local_mem_size
while True:
- lmem_size = wg*4*key_dtype.itemsize
+ lmem_size = wg*key_dtype.itemsize
if argsort:
- lmem_size += wg*4*idx_dtype.itemsize
+ lmem_size += wg*idx_dtype.itemsize
if lmem_size + 512 > available_lmem:
wg //= 2
--
GitLab