skip if < 12

Author: ksimpson-work

cuda_core/cuda/core/experimental/_memory.py

@@ -221,9 +221,6 @@ def __init__(self, reserved: bytes, size):
         self.reserved = reserved
         self._size = size
 
-    def resize(self, new_size):
-        raise NotImplementedError("Cannot resize an IPC buffer")
-
     def close(self):
         raise NotImplementedError("Cannot close an IPC buffer directly")
 
@@ -233,17 +230,12 @@ def copy_from(self, src, size=None):
     def copy_to(self, dst, size=None):
         raise NotImplementedError("Cannot copy from an IPC buffer")
 
-    def __del__(self):
-        # Override to prevent automatic closing
-        pass
-
     def __reduce__(self):
-        # Save only the reserved bytes, not the whole CUmemPoolPtrExportData object
+        # This is subject to change if the CumemPoolPtrExportData struct/object changes.
         return (self._reconstruct, (self.reserved, self._size))
 
     @classmethod
     def _reconstruct(cls, reserved, size):
-        # Create new instance
         instance = cls(reserved, size)
         return instance
 

cuda_core/tests/test_memory.py

@@ -22,7 +22,7 @@
 
 from cuda.core.experimental import Device
 from cuda.core.experimental._memory import Buffer, MemoryResource, Mempool
-from cuda.core.experimental._utils import handle_return
+from cuda.core.experimental._utils import get_binding_version, handle_return
 
 
 class DummyDeviceMemoryResource(MemoryResource):
@@ -296,6 +296,10 @@ def test_mempool():
 )
 def test_mempool_properties(property_name, expected_type):
     """Test all properties of the Mempool class."""
+    # skip test if cuda version is less than 12
+    if get_binding_version() < (12, 0):
+        pytest.skip("Test requires CUDA 12 or higher")
+
     device = Device()
     device.set_current()
     pool_size = 2097152  # 2MB size
@@ -392,6 +396,8 @@ def mempool_child_process(importer, queue):
 
 
 def test_ipc_mempool():
+    if get_binding_version() < (12, 0):
+        pytest.skip("Test requires CUDA 12 or higher")
     # Set multiprocessing start method before creating any multiprocessing objects
     multiprocessing.set_start_method("spawn", force=True)