Skip to content

Commit 1a5341f

Browse files
committed
compiler: add mode to IG creation, rebase on 2208
1 parent e96bb5e commit 1a5341f

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

devito/ir/support/space.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ def zero(self):
260260
return Interval(self.dim, 0, 0, self.stamp)
261261

262262
def ceil(self, o):
263-
if o.is_Null:
263+
if not self.is_compatible(o):
264264
return self._rebuild()
265265
return Interval(self.dim, self.lower, o.upper, self.stamp)
266266

@@ -399,7 +399,12 @@ def generate(cls, op, *interval_groups, relations=None):
399399
relations.update(set().union(*[ig.relations for ig in interval_groups]))
400400

401401
modes = set(ig.mode for ig in interval_groups)
402+
403+
# try:
402404
assert len(modes) <= 1
405+
# except:
406+
# import pdb;pdb.set_trace()
407+
403408
try:
404409
mode = modes.pop()
405410
except KeyError:
@@ -500,7 +505,7 @@ def zero(self, d=None):
500505
def ceil(self, o=None):
501506
d = self.dimensions if o is None else as_tuple(o.dim)
502507
return IntervalGroup([i.ceil(o) if i.dim in d else i for i in self],
503-
relations=self.relations)
508+
relations=self.relations, mode=self.mode)
504509

505510
def lift(self, d=None, v=None):
506511
d = set(self.dimensions if d is None else as_tuple(d))

0 commit comments

Comments
 (0)