Merge pull request #1748 from devitocodes/faster-glb-to-loc

mpi: Speedup index_glb_to_loc
devitocodes · Sep 5, 2021 · a5acfcb · a5acfcb
2 parents 18c842a + 7cfa884
commit a5acfcb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/devito/data/decomposition.py b/devito/data/decomposition.py
@@ -219,7 +219,7 @@ def index_glb_to_loc(self, *args, rel=True):
                 if glb_idx < 0:
                     glb_idx = self.glb_max + glb_idx + 1
                 # -> Do the actual conversion
-                if glb_idx in self.loc_abs_numb:
+                if self.loc_abs_min <= glb_idx <= self.loc_abs_max:
                     return glb_idx - base
                 elif self.glb_min <= glb_idx <= self.glb_max:
                     return None