compiler: Switch to alternative solution

devitocodes · Sep 4, 2023 · d36ba9b · d36ba9b
1 parent 368c792
commit d36ba9b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 11 deletions.
diff --git a/devito/ir/clusters/algorithms.py b/devito/ir/clusters/algorithms.py
@@ -209,14 +209,8 @@ def guard(clusters):
                 if cd._factor is not None:
                     k = d
                 else:
-                    dims = pull_dims(cd.condition)
-                    # If `cd` uses more dimensions than the ispace,
-                    # stay under parent
-                    if (not dims.issubset(set(c.ispace.dimensions)) and
-                       cd.parent in dims):
-                        k = cd.parent
-                    else:
-                        k = max(dims, default=d, key=lambda i: c.ispace.index(i))
+                    dims = pull_dims(cd.condition, flag=False)
+                    k = max(dims, default=d, key=lambda i: c.ispace.index(i))
 
                 # Pull `cd` from any expr
                 condition = guards.setdefault(k, [])

diff --git a/tests/test_subdomains.py b/tests/test_subdomains.py
@@ -632,7 +632,7 @@ class Dummy(SubDomainSet):
 
 class TestSubDomain_w_condition(object):
 
-    def test_condition_w_subdomain(self):
+    def test_condition_w_subdomain_v0(self):
 
         shape = (10, )
         grid = Grid(shape=shape)
@@ -663,7 +663,7 @@ def define(self, dimensions):
 
         assert_structure(op, ['i1x'], 'i1x')
 
-    def test_condition_w_subdomain_II(self):
+    def test_condition_w_subdomain_v1(self):
 
         shape = (10, 10)
         grid = Grid(shape=shape)
@@ -695,7 +695,7 @@ def define(self, dimensions):
 
         assert_structure(op, ['xi1y'], 'xi1y')
 
-    def test_condition_w_subdomain_III(self):
+    def test_condition_w_subdomain_v2(self):
 
         shape = (10, 10)
         grid = Grid(shape=shape)