Skip to content
Snippets Groups Projects

Fix Linear3D Indexing if Iteration Space Rank < 3

Merged Frederik Hennig requested to merge fhennig/fix-2d-gpu-indexing into v2.0-dev
2 files
+ 52
1
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -267,6 +267,11 @@ class GpuIndexing:
f" for a {rank}-dimensional kernel."
)
work_items_expr += tuple(
self._ast_factory.parse_index(1)
for _ in range(3 - rank)
)
num_work_items = cast(
_Dim3Lambda,
tuple(self._kernel_factory.create_lambda(wit) for wit in work_items_expr),
Loading