api: enforce sympy shifts

devito/finite_differences/tools.py

@@ -2,7 +2,7 @@
 from itertools import product
 
 import numpy as np
-from sympy import S, finite_diff_weights, cacheit, sympify, Function
+from sympy import S, finite_diff_weights, cacheit, sympify, Function, Rational
 
 from devito.tools import Tag, as_tuple
 from devito.types.dimension import StencilDimension
@@ -308,8 +308,8 @@ def make_shift_x0(shift, ndim):
     """
     if shift is None:
         return lambda s, d, i, j: None
-    elif isinstance(shift, float):
-        return lambda s, d, i, j: d + s * d.spacing
+    elif sympify(shift).is_Number:
+        return lambda s, d, i, j: d + Rational(s) * d.spacing
     elif type(shift) is tuple and np.shape(shift) == ndim:
         if len(ndim) == 1:
             return lambda s, d, i, j: d + s[j] * d.spacing

devito/symbolics/printer.py

@@ -7,6 +7,8 @@
 
 from mpmath.libmp import prec_to_dps, to_str
 from packaging.version import Version
+
+from sympy.codegen.ast import float32, float64
 from sympy.logic.boolalg import BooleanFunction
 from sympy.printing.precedence import PRECEDENCE_VALUES, precedence
 from sympy.printing.c import C99CodePrinter
@@ -18,6 +20,9 @@
 __all__ = ['ccode']
 
 
+_type_mapper = {np.float32: float32, np.float64: float64}
+
+
 class CodePrinter(C99CodePrinter):
 
     """
@@ -179,12 +184,20 @@ def _print_Add(self, expr, order=None):
 
     def _print_Float(self, expr):
         """Print a Float in C-like scientific notation."""
+        try:
+            # Make sure the float is in the correct format
+            expr = _type_mapper[self.dtype].cast_nocheck(expr)
+            rv = str(expr)
+        except KeyError:
+            pass
+
         prec = expr._prec
 
         if prec < 5:
             dps = 0
         else:
             dps = prec_to_dps(expr._prec)
+
         if self._settings["full_prec"] is True:
             strip = False
         elif self._settings["full_prec"] is False: