fix(allocations): ensure that allocations dict does not raise KeyError during finalization

Author: cpcloud

numba_cuda/numba/cuda/cudadrv/driver.py

@@ -1500,7 +1500,7 @@ def _alloc_finalizer(memory_manager, ptr, alloc_key, size):
 
     def core():
         if allocations:
-            del allocations[alloc_key]
+            allocations.pop(alloc_key, None)
         deallocations.add_item(driver.cuMemFree, ptr, size)
 
     return core