Skip to content

Commit 16aa70f

Browse files
committed
compiler: Enhance _alloc_object_on_low_lat_mem
1 parent 0eeff32 commit 16aa70f

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

devito/passes/iet/definitions.py

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

0 commit comments

Comments
 (0)