compiler: Patch double buffering

devitocodes · Oct 25, 2023 · 4706e66 · 4706e66
1 parent 522d475
commit 4706e66
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/devito/passes/clusters/buffering.py b/devito/passes/clusters/buffering.py
@@ -432,7 +432,11 @@ def __init_multi_buffering__(self):
 
         lhs, rhs = expr.args
 
-        self.xd = lhs.function.indices[self.dim]
+        maybe_xd = lhs.function.indices[self.dim]
+        if not isinstance(maybe_xd, CustomDimension):
+            maybe_xd = rhs.function.indices[self.dim]
+            assert isinstance(maybe_xd, CustomDimension)
+        self.xd = maybe_xd
 
         idx0 = lhs.indices[self.dim]
         idx1 = rhs.indices[self.dim]