compiler: Tweak lowering of HaloTouch

devitocodes · May 25, 2023 · 9a2c468 · 9a2c468
1 parent df820d6
commit 9a2c468
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/devito/ir/stree/algorithms.py b/devito/ir/stree/algorithms.py
@@ -146,7 +146,8 @@ def preprocess(clusters):
 
             found = []
             for c1 in list(queue):
-                if c1.halo_scheme.distributed_aindices & dims:
+                if c1.guards == c.guards and \
+                   c1.halo_scheme.distributed_aindices & dims:
                     found.append(c1)
                     queue.remove(c1)
 
@@ -155,6 +156,9 @@ def preprocess(clusters):
 
             processed.append(c.rebuild(syncs=syncs, halo_scheme=halo_scheme))
 
+    # Sanity check!
+    assert not queue
+
     return processed