Commit 85ed77f1 authored by Martin Bauer's avatar Martin Bauer
Browse files

Fix in GPU indexing

parent ca72adf4
......@@ -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))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment