diff options
Diffstat (limited to 'boltzgen/utility/printer.py')
-rw-r--r-- | boltzgen/utility/printer.py | 13 |
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' |