is_finalizing

Author: brandon-b-miller

cuda_core/cuda/core/experimental/_event.pyx

@@ -19,7 +19,7 @@ from cuda.core.experimental._utils.cuda_utils import (
     handle_return,
     is_shutting_down
 )
-
+import sys
 if TYPE_CHECKING:
     import cuda.bindings
     from cuda.core.experimental._device import Device
@@ -109,7 +109,7 @@ cdef class Event:
         self._ctx_handle = ctx_handle
         return self
 
-    cpdef safe_close(self, is_shutting_down=is_shutting_down):
+    cpdef safe_close(self, is_shutting_down=sys.is_finalizing):
         """Destroy the event."""
         if self._handle is not None:
             if not is_shutting_down():

cuda_core/cuda/core/experimental/_memory.pyx

@@ -9,7 +9,7 @@ from libc.stdint cimport uintptr_t
 from cuda.core.experimental._utils.cuda_utils cimport (
     _check_driver_error as raise_if_driver_error,
 )
-
+import sys
 import abc
 from typing import TypeVar, Union
 
@@ -59,13 +59,10 @@ cdef class Buffer:
     def __del__(self):
         self.safe_close()
 
-    cpdef safe_close(self, stream: Stream = None, is_shutting_down=is_shutting_down):
+    cpdef safe_close(self, stream: Stream = None, is_shutting_down=sys.is_finalizing):
         if self._ptr and self._mr is not None:
             if not is_shutting_down():
                 self._mr.deallocate(self._ptr, self._size, stream)
-            self._ptr = 0
-            self._mr = None
-            self._ptr_obj = None
 
     cpdef close(self, stream: Stream = None):
         """Deallocate this buffer asynchronously on the given stream.

cuda_core/cuda/core/experimental/_stream.pyx

@@ -8,6 +8,7 @@ from cuda.core.experimental._utils.cuda_utils cimport (
     _check_driver_error as raise_if_driver_error,
     check_or_create_options,
 )
+import sys
 
 import os
 import warnings
@@ -25,7 +26,6 @@ from cuda.core.experimental._utils.cuda_utils import (
     driver,
     get_device_from_ctx,
     handle_return,
-    is_shutting_down
 )
 
 
@@ -188,16 +188,13 @@ cdef class Stream:
     def __del__(self):
         self.safe_close()
 
-    cpdef safe_close(self, is_shutting_down=is_shutting_down):
+    cpdef safe_close(self, is_shutting_down=sys.is_finalizing):
         if self._owner is None:
             if self._handle and not self._builtin:
-                if not is_shutting_down():
+                if not is_shutting_down:
                     handle_return(driver.cuStreamDestroy(self._handle))
-        else:
-            self._owner = None
-        self._handle = None
 
-    cpdef close(self, is_shutting_down=is_shutting_down):
+    cpdef close(self):
         """Destroy the stream.
 
         Destroy the stream if we own it. Borrowed foreign stream

cuda_core/cuda/core/experimental/_utils/cuda_utils.pyx

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import atexit
 import functools
 import importlib.metadata
 from collections import namedtuple
@@ -223,25 +222,3 @@ def get_binding_version():
     except importlib.metadata.PackageNotFoundError:
         major_minor = importlib.metadata.version("cuda-python").split(".")[:2]
     return tuple(int(v) for v in major_minor)
-
-# This code is to signal when the interpreter is in shutdown mode
-# to prevent using globals that could be already deleted in
-# objects `__del__` method
-#
-# This solution is taken from the Numba/llvmlite code
-_shutting_down = [False]
-
-
-@atexit.register
-def _at_shutdown():
-    _shutting_down[0] = True
-
-
-def is_shutting_down(_shutting_down=_shutting_down):
-    """
-    Whether the interpreter is currently shutting down.
-    For use in finalizers, __del__ methods, and similar; it is advised
-    to early bind this function rather than look it up when calling it,
-    since at shutdown module globals may be cleared.
-    """
-    return _shutting_down[0]