From a1e1eeda3d5eb3a6db3082c4ec6cc0a1129e49c1 Mon Sep 17 00:00:00 2001 From: Tal Ben-Nun Date: Thu, 7 Nov 2024 18:02:12 -0800 Subject: [PATCH] Fix cppunparse for OpenCL types --- dace/codegen/cppunparse.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dace/codegen/cppunparse.py b/dace/codegen/cppunparse.py index edeb5270ca..e315254ad1 100644 --- 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: