mpi: fix halo ispace with subdomains

Author: mloubout

devito/ir/clusters/algorithms.py

@@ -472,7 +472,7 @@ def callback(self, clusters, prefix, seen=None):
             expr = Eq(self.B, HaloTouch(*points, halo_scheme=hs))
 
             key0 = lambda i: i in prefix[:-1] or i in hs.loc_indices  # noqa: B023
-            key1 = lambda i: i not in hs.distributed_defined  # noqa: B023
+            key1 = lambda i: not i._defines & set(hs.distributed_defined)  # noqa: B023
             key = lambda i: key0(i) and key1(i)  # noqa: B023
             ispace = c.ispace.project(key)
 

examples/seismic/self_adjoint/sa_03_iso_correctness.ipynb

@@ -578,9 +578,9 @@
      "output_type": "stream",
      "text": [
       "Operator `IsoFwdOperator` ran in 0.04 s\n",
-      "No source type defined, returning uninitiallized (zero) source\n",
+      "No source type defined, returning uninitialized (zero) source\n",
       "Operator `IsoAdjOperator` ran in 0.03 s\n",
-      "No source type defined, returning uninitiallized (zero) source\n",
+      "No source type defined, returning uninitialized (zero) source\n",
       "Operator `IsoAdjOperator` ran in 0.03 s\n"
      ]
     },
@@ -639,7 +639,7 @@
      "output_type": "stream",
      "text": [
       "Operator `IsoFwdOperator` ran in 0.03 s\n",
-      "No source type defined, returning uninitiallized (zero) source\n",
+      "No source type defined, returning uninitialized (zero) source\n",
       "Operator `IsoAdjOperator` ran in 0.03 s\n"
      ]
     },

examples/seismic/utils.py

@@ -194,7 +194,7 @@ def src(self):
     def new_src(self, name='src', src_type='self', coordinates=None):
         coords = coordinates or self.src_positions
         if self.src_type is None or src_type is None:
-            warning("No source type defined, returning uninitiallized (zero) source")
+            warning("No source type defined, returning uninitialized (zero) source")
             src = PointSource(name=name, grid=self.grid,
                               time_range=self.time_axis, npoint=self.nsrc,
                               coordinates=coords,