1 files changed, 13 insertions, 0 deletions
diff --git a/boltzgen/utility/printer.py b/boltzgen/utility/printer.py
new file mode 100644
index 0000000..b52a5e0
--- /dev/null
+++ b/boltzgen/utility/printer.py
@@ -0,0 +1,13 @@
+from sympy.printing.ccode import C99CodePrinter
+from sympy.codegen.ast import float32, float64
+
+class CudaCodePrinter(C99CodePrinter):
+    pass
+
+    def __init__(self, float_type, **args):
+        super(CudaCodePrinter, self).__init__(**args)
+        if float_type == 'float':
+            self.type_func_suffixes[float32]    = 'f'
+            self.type_func_suffixes[float64]    = 'f'
+            self.type_literal_suffixes[float32] = 'f'
+            self.type_literal_suffixes[float64] = 'f'