Skip to content

Commit 80fcbcb

Browse files
committed
compiler: Fix override oversight
1 parent 485455b commit 80fcbcb

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

devito/operator/operator.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1382,7 +1382,6 @@ def get_nbytes(obj):
13821382
and not i.is_ArrayBasic and not i.alias]
13831383

13841384
for i in op_symbols:
1385-
# Will overreport memory usage currently
13861385
try:
13871386
# TODO: is _obj even needed?
13881387
v = get_nbytes(self[i.name]._obj)
@@ -1467,7 +1466,6 @@ def nbytes_consumed_memmapped(self):
14671466

14681467
@cached_property
14691468
def nbytes_snapshots(self):
1470-
14711469
# Symbols in the operator which may or may not carry data
14721470
op_symbols = FindSymbols().visit(self.op)
14731471

@@ -1478,7 +1476,12 @@ def nbytes_snapshots(self):
14781476
disk = 0
14791477
for i in op_symbols:
14801478
try:
1481-
disk += i.size_snapshot*i._time_size_ideal*np.dtype(i.dtype).itemsize
1479+
v = self[i.name]._obj
1480+
except AttributeError:
1481+
v = self.get(i.name, i)
1482+
1483+
try:
1484+
disk += v.size_snapshot*v._time_size_ideal*np.dtype(v.dtype).itemsize
14821485
except AttributeError:
14831486
pass
14841487

0 commit comments

Comments
 (0)