compiler: fix various corner case of multi buffering

Author: mloubout

devito/ir/equations/equation.py

@@ -237,6 +237,14 @@ def __new__(cls, *args, **kwargs):
                     cond = d.relation(cond, GuardFactor(d))
                 conditionals[d] = cond
 
+        # Replace the ConditionalDimensions in `expr`
+        for d, cond in conditionals.items():
+            # Replace dimension with index
+            index = d.index
+            index = index - relational_min(cond, d.parent)
+            shift = relational_shift(cond, d.parent)
+            expr = uxreplace(expr, {d: IntDiv(index, d.symbolic_factor) + shift})
+
         # Merge conditionals when possible. E.g if we have an implicit_dim
         # and there is a dimension with the same parent, we ca merged
         # its condition
@@ -247,19 +255,13 @@ def __new__(cls, *args, **kwargs):
                 if cd.parent == d.parent and cd != d:
                     cond = conditionals.pop(d)
                     mode = cd.relation and d.relation
-                    conditionals[cd] = mode(cond, conditionals[cd])
+                    if issubclass(mode, sympy.Or):
+                        conditionals[d] = cond
+                        conditionals.pop(cd)
+                    else:
+                        conditionals[cd] = mode(cond, conditionals[cd])
                     break
 
-        conditionals = frozendict(conditionals)
-
-        # Replace the ConditionalDimensions in `expr`
-        for d, cond in conditionals.items():
-            # Replace dimension with index
-            index = d.index
-            index = index - relational_min(cond, d.parent)
-            shift = relational_shift(cond, d.parent)
-            expr = uxreplace(expr, {d: IntDiv(index, d.symbolic_factor) + shift})
-
         # Lower all Differentiable operations into SymPy operations
         rhs = diff2sympy(expr.rhs)
 

devito/passes/clusters/buffering.py

@@ -3,7 +3,7 @@
 from itertools import chain
 
 import numpy as np
-from sympy import S, simplify
+from sympy import Mod, S, simplify
 
 from devito.exceptions import CompilationError
 from devito.ir import (
@@ -203,7 +203,7 @@ def callback(self, clusters, prefix):
                     guards = c.guards
 
                 properties = c.properties.sequentialize(d)
-                if not isinstance(d, BufferDimension):
+                if not isinstance(d, BufferDimension) and c.guards[d].has(Mod):
                     properties = properties.prefetchable(d)
                 # `c` may be a HaloTouch Cluster, so with no vision of the `bdims`
                 properties = properties.parallelize(v.bdims).affine(v.bdims)
@@ -377,7 +377,12 @@ def generate_buffers(clusters, key, sregistry, options, **kwargs):
             buffer, = buffers
             xd = buffer.indices[dim]
         else:
-            size = infer_buffer_size(f, dim, clusters)
+            if len({c.guards[dim.root] for c in clusters}) > 1:
+                # Multiple clusters with different guards,
+                # will lead to conflicts in asynchrony with multiple (modulo) slots
+                size = 1
+            else:
+                size = infer_buffer_size(f, dim, clusters)
 
             if async_degree is not None:
                 if async_degree < size:

devito/symbolics/extended_sympy.py

@@ -18,7 +18,6 @@
 )
 from devito.types import Symbol
 from devito.types.basic import Basic
-from devito.types.relational import Ge
 
 __all__ = ['CondEq', 'CondNe', 'BitwiseNot', 'BitwiseXor', 'BitwiseAnd',  # noqa
            'LeftShift', 'RightShift', 'IntDiv', 'Terminal', 'CallFromPointer',
@@ -48,11 +47,6 @@ def canonical(self):
     def negated(self):
         return CondNe(*self.args, evaluate=False)
 
-    @property
-    def _as_min(self):
-        from devito.symbolics.extended_dtypes import INT
-        return INT(Ge(*self.args))
-
 
 class CondNe(sympy.Ne):
 

devito/types/relational.py

@@ -320,4 +320,5 @@ def _(expr, s):
 def _(expr, s):
     if isinstance(expr.lhs, sympy.Mod):
         return 0
-    return expr._as_min
+    from devito.symbolics.extended_dtypes import INT
+    return INT(Ge(*expr.args))