Skip to content

Commit d9b4086

Browse files
committed
compiler: Return useful information from ExprGeometry.translated
1 parent 661939b commit d9b4086

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

devito/ir/support/basic.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1266,6 +1266,7 @@ def translated(self, other, dims=None):
12661266
dims = set(as_tuple(dims))
12671267

12681268
# Check bases and offsets
1269+
distances = {}
12691270
for i in ['Tbases', 'Toffsets']:
12701271
Ti0 = getattr(self, i)
12711272
Ti1 = getattr(other, i)
@@ -1288,13 +1289,16 @@ def translated(self, other, dims=None):
12881289

12891290
distance = set(o0 - o1)
12901291
if len(distance) != 1:
1291-
return False
1292+
return {}
1293+
v = distance.pop()
12921294

12931295
if not d._defines & dims:
1294-
if distance.pop() != 0:
1295-
return False
1296+
if v != 0:
1297+
return {}
1298+
1299+
distances[d] = v
12961300

1297-
return True
1301+
return distances
12981302

12991303
@cached_property
13001304
def iinstances(self):

0 commit comments

Comments
 (0)