Skip to content
Snippets Groups Projects
Commit a100881c authored by Markus Holzer's avatar Markus Holzer
Browse files

Merge branch 'GPUBlockSize' into 'master'

Gpu block size

See merge request !296
parents 343d0708 47d0ca93
No related branches found
No related tags found
1 merge request!296Gpu block size
Pipeline #41155 passed
......@@ -147,7 +147,8 @@ class BlockIndexing(AbstractIndexing):
for i in range(len(widths)):
factor = div_floor(prod(block_size[:i]), prod(adapted_block_size))
adapted_block_size.append(sp.Min(block_size[i] * factor, widths[i]))
block_size = tuple(adapted_block_size) + extend_bs
extend_adapted_bs = (1,) * (3 - len(adapted_block_size))
block_size = tuple(adapted_block_size) + extend_adapted_bs
block_size = tuple(sp.Min(bs, max_bs) for bs, max_bs in zip(block_size, self._maximum_block_size))
grid = tuple(div_ceil(length, block_size) for length, block_size in zip(widths, block_size))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment