fix row->column major conversion

xdslproject · Oct 21, 2023 · 2079b9c · 2079b9c
1 parent 674081e
commit 2079b9c
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/devito/operator/xdsl_operator.py b/devito/operator/xdsl_operator.py
@@ -86,10 +86,8 @@ def _make_interop_o(self):
     @property
     def mpi_shape(self) -> tuple:
         dist = self.functions[0].grid.distributor
-        # temporary fix:
-        # swap dim 0 and 1 in topology because dmp.grid is row major and not column major
-
-        return (dist.topology[1], dist.topology[0], *dist.topology[2:]), dist.myrank
+        # reverse topology for row->column major
+        return tuple(reversed(dist.topology)), dist.myrank
 
     def _jit_compile(self):
         """