IPC Mempool Serialization and multiprocessing Module Support

cuda_core/cuda/core/experimental/__init__.py

@@ -17,7 +17,7 @@
 from cuda.core.experimental._memory import (
     Buffer,
     DeviceMemoryResource,
-    IPCChannel,
+    DeviceMemoryResourceOptions,
     LegacyPinnedMemoryResource,
     MemoryResource,
 )

cuda_core/cuda/core/experimental/_device.py

@@ -1160,6 +1160,9 @@ def __int__(self):
     def __repr__(self):
         return f"<Device {self._id} ({self.name})>"
 
+    def __reduce__(self):
+        return Device, (self.device_id,)
+
     def set_current(self, ctx: Context = None) -> Union[Context, None]:
         """Set device to be used for GPU executions.