Skip to content

Commit 19b825b

Browse files
committed
compiler: Fix oversight in refactored nbytes_avail_mapper
1 parent 0a4d71f commit 19b825b

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

devito/operator/operator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1328,7 +1328,10 @@ def nbytes_avail_mapper(self):
13281328
mapper[host_layer] = int(ANYCPU.memavail() / nproc)
13291329

13301330
for layer in (host_layer, device_layer):
1331-
mapper[layer] -= self.nbytes_consumed_operator.get(layer, 0)
1331+
try:
1332+
mapper[layer] -= self.nbytes_consumed_operator.get(layer, 0)
1333+
except KeyError: # Might not have this layer in the mapper
1334+
pass
13321335

13331336
mapper = {k: int(v) for k, v in mapper.items()}
13341337

0 commit comments

Comments
 (0)