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
Branches
Tags
1 merge request!296Gpu block size
Pipeline #41155 passed with stages
in 23 minutes and 54 seconds
...@@ -147,7 +147,8 @@ class BlockIndexing(AbstractIndexing): ...@@ -147,7 +147,8 @@ class BlockIndexing(AbstractIndexing):
for i in range(len(widths)): for i in range(len(widths)):
factor = div_floor(prod(block_size[:i]), prod(adapted_block_size)) factor = div_floor(prod(block_size[:i]), prod(adapted_block_size))
adapted_block_size.append(sp.Min(block_size[i] * factor, widths[i])) 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)) 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)) grid = tuple(div_ceil(length, block_size) for length, block_size in zip(widths, block_size))
......
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