hacky fix for row major dmp.grid

xdslproject · Aug 3, 2023 · ea7fe19 · ea7fe19
1 parent e4db7f9
commit ea7fe19
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/devito/operator/xdsl_operator.py b/devito/operator/xdsl_operator.py
@@ -85,7 +85,10 @@ def _make_interop_o(self):
     @property
     def mpi_shape(self) -> tuple:
         dist = self.functions[0].grid.distributor
-        return dist.topology, dist.myrank
+        # 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
 
     def _jit_compile(self):
         """