mpi: Cast MPI={1,True,basic} to MPI=basic for homogeneity

devitocodes · Sep 10, 2024 · eab3579 · eab3579
1 parent f314991
commit eab3579
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/devito/__init__.py b/devito/__init__.py
@@ -75,7 +75,7 @@ def reinit_compiler(val):
                   deprecate='openmp')
 
 # MPI mode (0 => disabled, 1 == basic)
-preprocessor = lambda i: bool(i) if isinstance(i, int) else i
+preprocessor = lambda i: {0: False, 1: 'basic'}.get(i, i)
 configuration.add('mpi', 0, [0, 1] + list(mpi_registry),
                   preprocessor=preprocessor, callback=reinit_compiler)
 

diff --git a/devito/mpi/routines.py b/devito/mpi/routines.py
@@ -1002,7 +1002,6 @@ def _call_poke(self, poke):
 
 
 mpi_registry = {
-    True: BasicHaloExchangeBuilder,
     'basic': BasicHaloExchangeBuilder,
     'diag': DiagHaloExchangeBuilder,
     'diag2': Diag2HaloExchangeBuilder,