Skip to content
Snippets Groups Projects

Fix width-one iteration slices on GPU

Merged Frederik Hennig requested to merge da15siwa/pystencils:fix-gpu-islices into master
Compare and
2 files
+ 91
13
Preferences
Compare changes
Files
2
@@ -66,7 +66,8 @@ def create_cuda_kernel(assignments: NodeCollection, config: CreateKernelConfig):
iteration_space = normalize_slice(iteration_slice, common_shape)
else:
iteration_space = normalize_slice(iteration_slice, common_shape)
iteration_space = tuple([s if isinstance(s, slice) else slice(s, s, 1) for s in iteration_space])
iteration_space = tuple([s if isinstance(s, slice) else slice(s, s + 1, 1) for s in iteration_space])
loop_counter_symbols = [LoopOverCoordinate.get_loop_counter_symbol(i) for i in range(len(iteration_space))]