Skip to content

Commit 70c6841

Browse files
committed
compiler: Hotfix estimate memory again
1 parent 767950b commit 70c6841

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

devito/operator/operator.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1364,8 +1364,10 @@ def _get_nbytes(self, i):
13641364
try:
13651365
# Non-regular AbstractFunction (compressed, etc)
13661366
nbytes = obj.nbytes_max
1367-
except AttributeError:
1368-
# Garden-variety AbstractFunction
1367+
except (AttributeError, ValueError):
1368+
# Either garden-variety AbstractFunction, or uninitialised
1369+
# function used in estimate. In the latter case, fall back
1370+
# to nbytes, as it is typically zero
13691371
nbytes = obj.nbytes
13701372

13711373
# Could nominally have symbolic nbytes at this point
@@ -1461,7 +1463,8 @@ def nbytes_consumed_arrays(self):
14611463
# Temporaries such as Arrays are allocated and deallocated on-the-fly
14621464
# while in C land, so they need to be accounted for as well
14631465
for i in self._op_symbols:
1464-
if not i.is_Array or not i._mem_heap or i.alias:
1466+
if not i.is_Array or not i._mem_heap or i.alias \
1467+
or not i.is_regular:
14651468
continue
14661469

14671470
if i.is_regular:

0 commit comments

Comments
 (0)