Extract target-dependent floating point type name

1 files changed, 15 insertions, 0 deletions

diff --git a/boltzgen/kernel/target/precision.py b/boltzgen/kernel/target/precision.py
new file mode 100644
index 0000000..45f84cf
--- /dev/null
+++ b/boltzgen/kernel/target/precision.py
@@ -0,0 +1,15 @@
+class cpp:
+    @classmethod
+    def get_float_type(self, precision):
+        return {
+            'single': 'float',
+            'double': 'double'
+        }.get(precision)
+
+class cl:
+    @classmethod
+    def get_float_type(self, precision):
+        return {
+            'single': 'float',
+            'double': 'double'
+        }.get(precision)