Fix cppunparse for OpenCL types

spcl · Nov 8, 2024 · a1e1eed · a1e1eed
1 parent 11f557e
commit a1e1eed
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dace/codegen/cppunparse.py b/dace/codegen/cppunparse.py
@@ -349,6 +349,8 @@ def _Assign(self, t):
                             # if the veclen is greater than one, this should be defined with a vector data type
                             self.write("{}{} ".format(dace.dtypes._OCL_VECTOR_TYPES[inferred_type.type],
                                                       inferred_type.veclen))
+                        elif self.language == dace.dtypes.Language.OpenCL:
+                            self.write(dace.dtypes._OCL_TYPES[inferred_type.type] + " ")
                         else:
                             self.write(dace.dtypes._CTYPES[inferred_type.type] + " ")
                     else: