diff --git a/devito/operator/xdsl_operator.py b/devito/operator/xdsl_operator.py index bc947d2fe67..223f7d7fa14 100644 --- a/devito/operator/xdsl_operator.py +++ b/devito/operator/xdsl_operator.py @@ -87,7 +87,7 @@ def _make_interop_o(self): def mpi_shape(self) -> tuple: dist = self.functions[0].grid.distributor # reverse topology for row->column major - return tuple(reversed(dist.topology)), dist.myrank + return dist.topology, dist.myrank def _jit_compile(self): """