Skip to content

Commit ea09060

Browse files
committed
compiler: Enhance _alloc_object_on_low_lat_mem
1 parent eab6282 commit ea09060

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

devito/passes/iet/definitions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,10 @@ def _alloc_object_on_low_lat_mem(self, site, obj, storage):
101101
if not init:
102102
definition = decl
103103
efuncs = ()
104+
elif isinstance(init, (list, tuple)):
105+
assert len(init) == 2, "Expected (efunc, call)"
106+
init, definition = init
107+
efuncs = (init,)
104108
elif init.is_Callable:
105109
definition = Call(init.name, init.parameters,
106110
retobj=obj if init.retval else None)

0 commit comments

Comments
 (0)