Skip to content
Snippets Groups Projects

Fix Linear3D Indexing if Iteration Space Rank < 3

2 files
+ 52
1
Compare changes
  • Side-by-side
  • Inline

Files

@@ -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