Rename target module to memory

1 files changed, 16 insertions, 0 deletions

diff --git a/boltzgen/kernel/memory/precision.py b/boltzgen/kernel/memory/precision.py
new file mode 100644
index 0000000..d22e922
--- /dev/null
+++ b/boltzgen/kernel/memory/precision.py
@@ -0,0 +1,16 @@
+class common:
+    @classmethod
+    def get_float_type(self, precision):
+        if precision not in ['single', 'double']:
+            raise Exception("Precision must be either 'single' or 'double'")
+
+        return {
+            'single': 'float',
+            'double': 'double'
+        }.get(precision)
+
+class cpp(common):
+    pass
+
+class cl(common):
+    pass