merge mempool classes

Author: ksimpson-work

cuda_core/cuda/core/experimental/__init__.py

@@ -7,7 +7,7 @@
 from cuda.core.experimental._event import EventOptions
 from cuda.core.experimental._launcher import LaunchConfig, launch
 from cuda.core.experimental._linker import Linker, LinkerOptions
-from cuda.core.experimental._memory import Mempool
+from cuda.core.experimental._memory import AsyncMempool
 from cuda.core.experimental._module import ObjectCode
 from cuda.core.experimental._program import Program, ProgramOptions
 from cuda.core.experimental._stream import Stream, StreamOptions

cuda_core/cuda/core/experimental/_device.py

@@ -6,7 +6,7 @@
 from typing import Union
 
 from cuda.core.experimental._context import Context, ContextOptions
-from cuda.core.experimental._memory import Buffer, MemoryResource, _DefaultAsyncMempool, _SynchronousMemoryResource
+from cuda.core.experimental._memory import AsyncMempool, Buffer, MemoryResource, _SynchronousMemoryResource
 from cuda.core.experimental._stream import Stream, StreamOptions, default_stream
 from cuda.core.experimental._utils import ComputeCapability, CUDAError, driver, handle_return, precondition, runtime
 
@@ -962,7 +962,7 @@ def __new__(cls, device_id=None):
                             runtime.cudaDeviceGetAttribute(runtime.cudaDeviceAttr.cudaDevAttrMemoryPoolsSupported, 0)
                         )
                     ) == 1:
-                        dev._mr = _DefaultAsyncMempool(dev_id)
+                        dev._mr = AsyncMempool._from_device(dev_id)
                     else:
                         dev._mr = _SynchronousMemoryResource(dev_id)