We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0a4d71f commit 19b825bCopy full SHA for 19b825b
1 file changed
devito/operator/operator.py
@@ -1328,7 +1328,10 @@ def nbytes_avail_mapper(self):
1328
mapper[host_layer] = int(ANYCPU.memavail() / nproc)
1329
1330
for layer in (host_layer, device_layer):
1331
- mapper[layer] -= self.nbytes_consumed_operator.get(layer, 0)
+ try:
1332
+ mapper[layer] -= self.nbytes_consumed_operator.get(layer, 0)
1333
+ except KeyError: # Might not have this layer in the mapper
1334
+ pass
1335
1336
mapper = {k: int(v) for k, v in mapper.items()}
1337
0 commit comments