compiler: patch multi-subdim isapce handling

Author: mloubout

devito/finite_differences/tools.py

@@ -4,7 +4,6 @@
 import numpy as np
 from sympy import S, finite_diff_weights, cacheit, sympify, Rational, Expr
 
-from devito.logger import warning
 from devito.tools import Tag, as_tuple
 from devito.types.dimension import StencilDimension
 
@@ -269,9 +268,7 @@ def generate_indices(expr, dim, order, side=None, matvec=None, x0=None, nweights
                              f"stencil size ({order + 1}) for order {order} scheme")
         elif do > dw:
             order = nweights - nweights % 2
-            warning(f"Less weights ({nweights}) provided than the stencil size"
-                    f"({order + 1}) for order {order} scheme."
-                    f" Reducing order to {order}")
+
     # Evaluation point
     x0 = sympify(((x0 or {}).get(dim) or expr.indices_ref[dim]))
 

devito/passes/clusters/implicit.py

@@ -175,13 +175,17 @@ def callback(self, clusters, prefix):
 
             # Make sure the "implicit expressions" are scheduled in
             # the innermost loop such that the thicknesses can be computed
-            edims = set(retrieve_dimensions(mapper.values(), deep=True))
-            if dim not in edims or not edims.issubset(prefix.dimensions):
+            def key(tkn):
+                edims = set(retrieve_dimensions(tkn, deep=True))
+                return dim._defines & edims and edims.issubset(prefix.dimensions)
+
+            mapper = {k: v for k, v in mapper.items() if key(v)}
+            if not mapper:
                 continue
 
             found[d.functions].clusters.append(c)
             found[d.functions].mapper = reduce(found[d.functions].mapper,
-                                               mapper, edims, prefix)
+                                               mapper, {dim}, prefix)
 
         # Turn the reduced mapper into a list of equations
         processed = []
@@ -262,7 +266,7 @@ def reduce(m0, m1, edims, prefix):
     def key(i):
         try:
             return i.indices[d]
-        except AttributeError:
+        except (KeyError, AttributeError):
             return i
 
     mapper = {}

devito/passes/iet/misc.py

@@ -6,7 +6,6 @@
 
 from devito.finite_differences import Max, Min
 from devito.finite_differences.differentiable import SafeInv
-from devito.logger import warning
 from devito.ir import (Any, Forward, DummyExpr, Iteration, EmptyList, Prodder,
                        FindApplications, FindNodes, FindSymbols, Transformer,
                        Uxreplace, filter_iterations, retrieve_iteration_tree,
@@ -155,7 +154,7 @@ def _generate_macros(iet, tracker=None, langbb=None, printer=CPrinter, **kwargs)
                      for define, expr in headers)
 
     # Generate Macros from higher-level SymPy objects
-    mheaders, includes = _generate_macros_math(iet, langbb=langbb)
+    mheaders, includes = _generate_macros_math(iet, langbb=langbb, printer=printer)
     includes = sorted(includes, key=str)
     headers.extend(sorted(mheaders, key=str))
 
@@ -199,25 +198,25 @@ def _generate_macros_findexeds(iet, sregistry=None, tracker=None, **kwargs):
     return iet
 
 
-def _generate_macros_math(iet, langbb=None):
+def _generate_macros_math(iet, langbb=None, printer=CPrinter):
     headers = []
     includes = []
     for i in FindApplications().visit(iet):
-        header, include = _lower_macro_math(i, langbb)
+        header, include = _lower_macro_math(i, langbb, printer)
         headers.extend(header)
         includes.extend(include)
 
     return headers, set(includes) - {None}
 
 
 @singledispatch
-def _lower_macro_math(expr, langbb):
+def _lower_macro_math(expr, langbb, printer):
     return (), {}
 
 
 @_lower_macro_math.register(Min)
 @_lower_macro_math.register(sympy.Min)
-def _(expr, langbb):
+def _(expr, langbb, printer):
     if has_integer_args(*expr.args):
         return (('MIN(a,b)', ('(((a) < (b)) ? (a) : (b))')),), {}
     else:
@@ -226,23 +225,25 @@ def _(expr, langbb):
 
 @_lower_macro_math.register(Max)
 @_lower_macro_math.register(sympy.Max)
-def _(expr, langbb):
+def _(expr, langbb, printer):
     if has_integer_args(*expr.args):
         return (('MAX(a,b)', ('(((a) > (b)) ? (a) : (b))')),), {}
     else:
         return (), as_tuple(langbb.get('header-math'))
 
 
 @_lower_macro_math.register(SafeInv)
-def _(expr, langbb):
+def _(expr, langbb, printer):
     try:
-        eps = np.finfo(expr.base.dtype).resolution**2
-    except ValueError:
-        warning(f"dtype not recognized in SafeInv for {expr.base}, assuming float32")
-        eps = np.finfo(np.float32).resolution**2
-    b = Cast('b', dtype=np.float32)
+        dtype = expr.base.dtype
+    except (AttributeError, ValueError):
+        dtype = np.float32
+    eps = np.finfo(dtype).resolution**2
+    b = printer()._print(Cast('b', dtype=dtype))
+    ext = 'F' if dtype is np.float32 else ''
     return (('SAFEINV(a, b)',
-             f'(((a) < {eps}F || ({b}) < {eps}F) ? (0.0F) : ((1.0F) / (a)))'),), {}
+             f'(((a) < {eps}{ext} || ({b}) < {eps}{ext}) ? '
+             f'(0.0{ext}) : ((1.0{ext}) / (a)))'),), {}
 
 
 @iet_pass

devito/symbolics/inspection.py

@@ -13,7 +13,7 @@
 from devito.symbolics.extended_sympy import (CallFromPointer, Cast,
                                              DefFunction, ReservedWord)
 from devito.symbolics.queries import q_routine
-from devito.tools import as_tuple, prod
+from devito.tools import as_tuple, prod, is_integer
 from devito.tools.dtypes_lowering import infer_dtype
 
 __all__ = ['compare_ops', 'estimate_cost', 'has_integer_args', 'sympy_dtype']
@@ -287,13 +287,17 @@ def has_integer_args(*args):
         try:
             return np.issubdtype(args[0].dtype, np.integer)
         except AttributeError:
-            return args[0].is_integer
+            return is_integer(args[0])
 
     res = True
     for a in args:
         try:
             if isinstance(a, INT):
                 res = res and True
+            elif is_integer(a):
+                res = res and True
+            elif has_integer_args(a):
+                res = res and True
             elif len(a.args) > 0:
                 res = res and has_integer_args(*a.args)
             else: