converto scalar to array on nsdfg output

GridTools · Dec 10, 2024 · 746f9d8 · 746f9d8
1 parent 9bdc75b
commit 746f9d8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/simplify.py b/src/gt4py/next/program_processors/runners/dace_fieldview/transformations/simplify.py
@@ -996,6 +996,10 @@ def apply(
             else:
                 assert isinstance(inner_desc, dace.data.Scalar)
                 assert len(new_strides) == 0
+                # we convert the scalar data to array to avoid a gpu codegen error
+                nsdfg_node.sdfg.arrays[inner_data] = dace.data.Array(
+                    inner_desc.dtype, (1,), inner_desc.transient
+                )
             for stride in new_strides:
                 for sym in stride.free_symbols:
                     nsdfg_node.sdfg.add_symbol(str(sym), sym.dtype)