add resource management to mempool

Author: ksimpson-work

cuda_core/cuda/core/experimental/_memory.py

@@ -251,34 +251,11 @@ def _reconstruct(cls, reserved, size):
 class MemoryResource(abc.ABC):
     """Base class for memory resources.
 
-    This class provides an abstract interface for memory resources and includes
-    an optional destruction mechanism through _MembersNeededForFinalize.
+    This class provides an abstract interface for memory resources.
     """
 
-    class _MembersNeededForFinalize:
-        __slots__ = ("handle", "is_initialized")
-
-        def __init__(self, mr_obj, handle=None):
-            self.handle = handle
-            self.is_initialized = False
-            weakref.finalize(mr_obj, self.close)
-
-        def close(self):
-            if self.is_initialized and self.handle:
-                # Specific cleanup can be implemented by derived classes
-                self.handle = None
-                self.is_initialized = False
-
-    __slots__ = ("__weakref__", "_mnff")
-
     @abc.abstractmethod
-    def __init__(self, *args, **kwargs):
-        self._mnff = MemoryResource._MembersNeededForFinalize(self)
-
-    def close(self):
-        """Release any resources associated with this memory resource."""
-        if hasattr(self, "_mnff"):
-            self._mnff.close()
+    def __init__(self, *args, **kwargs): ...
 
     @abc.abstractmethod
     def allocate(self, size, stream=None) -> Buffer: ...
@@ -329,29 +306,6 @@ class Mempool(MemoryResource):
     Use the static methods `create` or `from_shared_handle` to instantiate a Mempool.
     Direct instantiation is not supported.
 
-    Properties
-    ----------
-    reuse_follow_event_dependencies : bool
-        Allow memory to be reused when there are event dependencies between streams.
-    reuse_allow_opportunistic : bool
-        Allow reuse of completed frees without dependencies.
-    reuse_allow_internal_dependencies : bool
-        Allow insertion of new stream dependencies for memory reuse.
-    release_threshold : int
-        Amount of reserved memory to hold before OS release.
-    reserved_mem_current : int
-        Current amount of backing memory allocated.
-    reserved_mem_high : int
-        High watermark of backing memory allocated.
-    used_mem_current : int
-        Current amount of memory in use.
-    used_mem_high : int
-        High watermark of memory in use.
-
-    See Also
-    --------
-    create : Create a new memory pool
-    from_shared_handle : Import an existing memory pool from another process
     """
 
     class _MembersNeededForFinalize: