Skip to content

Commit 4a959f6

Browse files
committed
WIP WIP CHANGING defines-aliases to defines in LINEARIZE WARNING
1 parent 6036e1f commit 4a959f6

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

devito/passes/iet/linearization.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def linearize_accesses(iet, key0, tracker=None):
223223

224224
# 2) What `iet` *offers*
225225
# E.g. `{x_fsz0 -> u_vec->size[1]}`
226-
defines = FindSymbols('defines-aliases').visit(iet)
226+
defines = FindSymbols('defines').visit(iet)
227227
offers = filter_ordered(i for i in defines if key0(i.function))
228228
instances = {}
229229
for i in offers:
@@ -294,16 +294,9 @@ def _(f, d):
294294

295295

296296
@_generate_fsz.register(Array)
297-
def _(f, d):
298-
return f.symbolic_shape[d]
299-
300-
301297
@_generate_fsz.register(Bundle)
302298
def _(f, d):
303-
if f.is_DiscreteFunction:
304-
return _generate_fsz.registry[DiscreteFunction](f, d)
305-
else:
306-
return _generate_fsz.registry[Array](f, d)
299+
return f.symbolic_shape[d]
307300

308301

309302
@singledispatch

0 commit comments

Comments
 (0)