diff --git a/pystencils/backends/cbackend.py b/pystencils/backends/cbackend.py index 18fd1d344b750b61bd24b1ddfc45199da14848f2..cd71d0ffb3ccc0e7c54bea2a7a08477fb019f451 100644 --- a/pystencils/backends/cbackend.py +++ b/pystencils/backends/cbackend.py @@ -13,7 +13,7 @@ except ImportError: from sympy.printing.ccode import CCodePrinter # for sympy versions < 1.1 from pystencils.integer_functions import bitwise_xor, bit_shift_right, bit_shift_left, bitwise_and, \ - bitwise_or, modulo_ceil, int_div, int_power_of_2 + bitwise_or, modulo_ceil, int_div, int_power_of_2, int_mod from pystencils.astnodes import Node, KernelFunction from pystencils.data_types import create_type, PointerType, get_type_of_expression, VectorType, cast_func, \ vector_memory_access, reinterpret_cast_func @@ -308,6 +308,8 @@ class CustomSympyPrinter(CCodePrinter): return "(1 << (%s))" % (self._print(expr.args[0])) elif expr.func == int_div: return "((%s) / (%s))" % (self._print(expr.args[0]), self._print(expr.args[1])) + elif expr.func == int_mod: + return "((%s) %% (%s))" % (self._print(expr.args[0]), self._print(expr.args[1])) else: return super(CustomSympyPrinter, self)._print_Function(expr) diff --git a/pystencils/integer_functions.py b/pystencils/integer_functions.py index 2fbc1ed673a230112ff7266922a7dc0f06a0eba2..5b243622eb6d1708d1fda514d66e0f6eec47a020 100644 --- a/pystencils/integer_functions.py +++ b/pystencils/integer_functions.py @@ -10,6 +10,7 @@ bitwise_and = sp.Function("bitwise_and") bitwise_or = sp.Function("bitwise_or") int_div = sp.Function("int_div") int_power_of_2 = sp.Function("int_power_of_2") +int_mod = sp.Function("int_mod") # noinspection PyPep8Naming