Fix upper shape

SeisSol · Apr 6, 2024 · 6cca2a1 · 6cca2a1
1 parent 8371ea8
commit 6cca2a1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/yateto/codegen/gpukernel.py b/yateto/codegen/gpukernel.py
@@ -69,7 +69,9 @@ def make_tensor(op, dims):
           for i, dim in enumerate(dims):
             if dim == 0:
               currentRange[i] = currentRange[i].aligned(self._arch)
-              currentShape[i] = max(currentShape[i], currentRange[i].stop)
+
+              # unstable/incorrect? TODO: check (for now, it should work)
+              currentShape[i] = max(self._arch.alignedUpper(currentShape[i]), currentRange[i].stop)
         entry = self._get_kernelforge_matrix(tensor=op,
                                             tensor_variable=op,
                                             shape=currentShape,