From 70eebe5a310afbcbd7894f9b8b6b0c7f92e0327c Mon Sep 17 00:00:00 2001
From: Anton Lydike <me@antonlydike.de>
Date: Wed, 18 Oct 2023 11:04:45 +0100
Subject: [PATCH] fix row->column major conversion

---
 devito/operator/xdsl_operator.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/devito/operator/xdsl_operator.py b/devito/operator/xdsl_operator.py
index a9a11a04af7..bc947d2fe67 100644
--- 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):
         """