fix use of wrong var

xdslproject · Oct 18, 2023 · c76773b · c76773b
1 parent 01964a7
commit c76773b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/devito/operator/xdsl_operator.py b/devito/operator/xdsl_operator.py
@@ -135,7 +135,7 @@ def _jit_compile(self):
                 # reduce the domain of the computation (as devito has already done that for us)
                 slices = ','.join(str(x) for x in shape)
 
-                decomp = "2d-grid" if len(slices) == 2 else "3d-grid"
+                decomp = "2d-grid" if len(shape) == 2 else "3d-grid"
 
                 decomp = f"{{strategy={decomp} slices={slices} restrict_domain=false}}"
                 xdsl_pipeline = XDSL_MPI_PIPELINE(decomp, to_tile)