api: cleanup based on first review

devito/ir/clusters/cluster.py

@@ -333,7 +333,6 @@ def dspace(self):
 
         # Construct the `intervals` of the DataSpace, that is a global,
         # Dimension-centric view of the data space
-
         intervals = IntervalGroup.generate('union', *parts.values())
         # E.g., `db0 -> time`, but `xi NOT-> x`
         intervals = intervals.promote(lambda d: not d.is_Sub)

devito/ir/equations/algorithms.py

@@ -56,9 +56,7 @@ def handle_indexed(indexed):
     # such as A[3])
     indexeds = retrieve_indexed(expr, deep=True)
     for i in indexeds:
-        expl_dims = {d for (d, e) in zip(i.function.dimensions, i.indices)
-                     if e.is_integer}
-        extra.update(expl_dims)
+        extra.update({d for d in i.function.dimensions if i.indices[d].is_integer})
 
     # Enforce determinism
     extra = filter_sorted(extra)

devito/ir/support/basic.py

@@ -660,8 +660,9 @@ def is_const(self, dim):
         """
         True if a constant dependence, that is no Dimensions involved, False otherwise.
         """
-        return (self.source.aindices.get(dim, None) is None and
-                self.sink.aindices.get(dim, None) is None and
+        print(self.source.aindices, dim)
+        return (self.source.aindices.get(dim) is None and
+                self.sink.aindices.get(dim) is None and
                 self.distance_mapper.get(dim, 0) == 0)
 
     @memoized_meth

devito/operations/interpolators.py

@@ -141,7 +141,7 @@ def _weights(self):
         raise NotImplementedError
 
     @property
-    def _gdim(self):
+    def _gdims(self):
         return self.grid.dimensions
 
     @property
@@ -153,9 +153,9 @@ def _rdim(self):
         parent = self.sfunction.dimensions[-1]
         dims = [CustomDimension("r%s%s" % (self.sfunction.name, d.name),
                                 -self.r+1, self.r, 2*self.r, parent)
-                for d in self._gdim]
+                for d in self._gdims]
 
-        return DimensionTuple(*dims, getters=self._gdim)
+        return DimensionTuple(*dims, getters=self._gdims)
 
     def _augment_implicit_dims(self, implicit_dims):
         return as_tuple(implicit_dims) + self.sfunction.dimensions
@@ -178,7 +178,7 @@ def _interp_idx(self, variables, implicit_dims=None):
 
         # Coefficient symbol expression
         temps.extend(self._coeff_temps(implicit_dims))
-        for ((di, d), rd, p) in zip(enumerate(self._gdim), self._rdim, pos):
+        for ((di, d), rd, p) in zip(enumerate(self._gdims), self._rdim, pos):
             # Add conditional to avoid OOB
             lb = sympy.And(rd + p >= d.symbolic_min - self.r, evaluate=False)
             ub = sympy.And(rd + p <= d.symbolic_max + self.r, evaluate=False)
@@ -330,7 +330,7 @@ class LinearInterpolator(WeightedInterpolator):
     @property
     def _weights(self):
         c = [(1 - p) * (1 - r) + p * r
-             for (p, d, r) in zip(self._point_symbols, self._gdim, self._rdim)]
+             for (p, d, r) in zip(self._point_symbols, self._gdims, self._rdim)]
         return Mul(*c)
 
     @cached_property
@@ -345,7 +345,7 @@ def _coeff_temps(self, implicit_dims):
         pmap = self.sfunction._position_map
         poseq = [Eq(self._point_symbols[d], pos - floor(pos),
                     implicit_dims=implicit_dims)
-                 for (d, pos) in zip(self._gdim, pmap.keys())]
+                 for (d, pos) in zip(self._gdims, pmap.keys())]
         return poseq
 
 

devito/passes/clusters/aliases.py

@@ -837,7 +837,7 @@ def lower_schedule(schedule, meta, sregistry, ftemps):
         # This prevents cases such as `floor(a*b)` with `a` and `b` floats
         # that would creat a temporary `int r = b` leading to erronous numerical results
         # Such cases happen with the positions for sparse functions for example.
-        dtype = sympy_dtype(pivot, meta.dtype) or meta.dtype
+        dtype = sympy_dtype(pivot, meta.dtype)
 
         if writeto:
             # The Dimensions defining the shape of Array

devito/passes/iet/mpi.py

@@ -47,9 +47,11 @@ def _drop_halospots(iet):
 
     # If a HaloSpot is outside any iteration it is not needed
     for iters, halo_spots in MapNodes(Iteration, HaloSpot, 'groupby').visit(iet).items():
+        if iters:
+            continue
         for hs in halo_spots:
             for f, v in hs.fmapper.items():
-                if not iters and v.loc_indices:
+                if v.loc_indices:
                     mapper[hs].add(f)
 
     # Transform the IET introducing the "reduced" HaloSpots

devito/passes/iet/parpragma.py

@@ -426,7 +426,12 @@ def _make_guard(self, parregion):
 
     def _make_nested_partree(self, partree):
         # Apply heuristic
-        if self.nhyperthreads <= self.nested or partree.root.is_ParallelAtomic:
+        if self.nhyperthreads <= self.nested:
+            return partree
+
+        # Loop nest with atomic reductions are more likely to have less latency
+        # keep outer loop parallel
+        if partree.root.is_ParallelAtomic:
             return partree
 
         # Note: there might be multiple sub-trees amenable to nested parallelism,

devito/symbolics/inspection.py

@@ -281,4 +281,5 @@ def sympy_dtype(expr, default):
             return default
         else:
             # Infer expression dtype from its arguments
-            return infer_dtype([sympy_dtype(a, default) for a in expr.args])
+            dtype = infer_dtype([sympy_dtype(a, default) for a in expr.args])
+            return dtype or default

devito/tools/data_structures.py

@@ -66,7 +66,7 @@ def __getnewargs_ex__(self):
         # objects with varying number of attributes
         return (tuple(self), dict(self.__dict__))
 
-    def get(self, key, val):
+    def get(self, key, val=None):
         return self._getters.get(key, val)
 
 
@@ -605,6 +605,7 @@ class UnboundTuple(object):
     """
     A simple data structure that returns the last element forever once reached
     """
+
     def __init__(self, items):
         self.items = as_tuple(items)
         self.last = len(self.items)

devito/types/dense.py

@@ -1460,9 +1460,7 @@ def parent(self):
 
     @property
     def origin(self):
-        """
-        SubFunction have zero origin
-        """
+        # SubFunction have zero origin
         return DimensionTuple(*(0 for _ in range(self.ndim)), getters=self.dimensions)