From 85ed77f1528446ab13503b0a671d2037654f0e9f Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 25 Oct 2018 16:50:26 +0200
Subject: [PATCH] Fix in GPU indexing
---
gpucuda/indexing.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/gpucuda/indexing.py b/gpucuda/indexing.py
index 979083c7e..85de69860 100644
--- a/gpucuda/indexing.py
+++ b/gpucuda/indexing.py
@@ -108,7 +108,7 @@ class BlockIndexing(AbstractIndexing):
extend_bs = (1,) * (3 - len(self._block_size))
block_size = self._block_size + extend_bs
if not self._compile_time_block_size:
- block_size = [sp.Min(bs, shape) for bs, shape in zip(block_size, widths)]
+ block_size = tuple(sp.Min(bs, shape) for bs, shape in zip(block_size, widths)) + extend_bs
grid = tuple(div_ceil(length, block_size)
for length, block_size in zip(widths, block_size))
--
GitLab