Skip to content

Commit

Permalink
compiler: Tidy up Interval.expand()
Browse files Browse the repository at this point in the history
  • Loading branch information
FabioLuporini committed Feb 6, 2023
1 parent a78cd4b commit b528c17
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 11 deletions.
1 change: 0 additions & 1 deletion devito/ir/clusters/cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,7 +255,6 @@ def dspace(self):
if f is None:
continue

#TODO OOOOOOOo
intervals = [Interval(d,
min([sdims_min(i) for i in offs]),
max([sdims_max(i) for i in offs]))
Expand Down
14 changes: 4 additions & 10 deletions devito/ir/support/space.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,11 @@
from cached_property import cached_property
from sympy import Expr

from devito.ir.support.utils import sdims_min, sdims_max
from devito.ir.support.vector import Vector, vmin, vmax
from devito.tools import (PartialOrderTuple, Stamp, as_list, as_tuple, filter_ordered,
flatten, frozendict, is_integer, toposort)
from devito.types import Dimension, ModuloDimension, StencilDimension
from devito.types import Dimension, ModuloDimension

__all__ = ['NullInterval', 'Interval', 'IntervalGroup', 'IterationSpace',
'DataSpace', 'Forward', 'Backward', 'Any']
Expand Down Expand Up @@ -285,15 +286,8 @@ def promote(self, cond):
return self

def expand(self):
lower = self.lower
if isinstance(self.lower, Expr):
for i in self.lower.find(StencilDimension):
lower = lower.subs(i, i._min)
upper = self.upper
if isinstance(self.upper, Expr):
for i in self.upper.find(StencilDimension):
upper = upper.subs(i, i._max)
return Interval(self.dim, lower, upper, self.stamp)
return Interval(self.dim, sdims_min(self.lower), sdims_max(self.upper),
self.stamp)


class IntervalGroup(PartialOrderTuple):
Expand Down

0 comments on commit b528c17

Please sign in to comment.