Add Python bindings for SYCL IPC memory via dpctl

CMakeLists.txt

@@ -115,7 +115,7 @@ install(DIRECTORY
 )
 
 # find Python before enabling pybind11
-find_package(Python REQUIRED COMPONENTS Development.Module)
+find_package(Python REQUIRED COMPONENTS Interpreter Development.Module)
 
 # Define CMAKE_INSTALL_xxx: LIBDIR, INCLUDEDIR
 include(GNUInstallDirs)

dpctl/_backend.pxd

@@ -101,6 +101,7 @@ cdef extern from "syclinterface/dpctl_sycl_enum_types.h":
         _emulated                           "emulated",
         _is_component                       "is_component",
         _is_composite                       "is_composite",
+        _ext_oneapi_ipc_memory              "ext_oneapi_ipc_memory",
 
     ctypedef enum _partition_affinity_domain_type \
             "DPCTLPartitionAffinityDomainType":
@@ -595,6 +596,7 @@ cdef extern from "syclinterface/dpctl_sycl_extension_interface.h":
         DPCTLSyclWorkGroupMemoryRef Ref)
 
     cdef bint DPCTLWorkGroupMemory_Available()
+    cdef bint DPCTLIPCMem_Available()
 
     cdef struct DPCTLOpaqueRawKernelArg
     ctypedef DPCTLOpaqueRawKernelArg *DPCTLSyclRawKernelArgRef
@@ -606,3 +608,19 @@ cdef extern from "syclinterface/dpctl_sycl_extension_interface.h":
         DPCTLSyclRawKernelArgRef Ref)
 
     cdef bint DPCTLRawKernelArg_Available()
+
+cdef extern from "syclinterface/dpctl_sycl_ipc_memory_interface.h":
+    cdef int DPCTLIPCMem_GetHandle(
+        DPCTLSyclUSMRef Ptr,
+        DPCTLSyclContextRef CRef,
+        char **DataOut,
+        size_t *SizeOut)
+    cdef DPCTLSyclUSMRef DPCTLIPCMem_OpenHandle(
+        const char *HandleData,
+        size_t HandleDataSize,
+        DPCTLSyclContextRef CRef,
+        DPCTLSyclDeviceRef DRef)
+    cdef void DPCTLIPCMem_CloseHandle(
+        DPCTLSyclUSMRef MappedPtr,
+        DPCTLSyclContextRef CRef)
+    cdef void DPCTLIPCMem_FreeHandleData(char *Data)

dpctl/_sycl_device.pyx

@@ -885,6 +885,18 @@ cdef class SyclDevice(_SyclDevice):
         cdef _aspect_type AT = _aspect_type._is_composite
         return DPCTLDevice_HasAspect(self._device_ref, AT)
 
+    @property
+    def has_aspect_ext_oneapi_ipc_memory(self):
+        """ Returns ``True`` if this device supports inter-process
+        communication (IPC) memory handles, ``False`` otherwise.
+
+        Returns:
+            bool:
+                Indicates if device supports IPC memory.
+        """
+        cdef _aspect_type AT = _aspect_type._ext_oneapi_ipc_memory
+        return DPCTLDevice_HasAspect(self._device_ref, AT)
+
     @property
     def image_2d_max_width(self):
         """ Returns the maximum width of a 2D image or 1D image in pixels.

dpctl/memory/__init__.py

@@ -30,17 +30,25 @@
 """
 
 from ._memory import (
+    IPCMemoryHandle,
     MemoryUSMDevice,
     MemoryUSMHost,
     MemoryUSMShared,
     USMAllocationError,
     as_usm_memory,
+    SyclIpcGetMemHandle,
+    SyclIpcOpenMemHandle,
+    SyclIpcCloseMemHandle,
 )
 
 __all__ = [
+    "IPCMemoryHandle",
     "MemoryUSMDevice",
     "MemoryUSMHost",
     "MemoryUSMShared",
     "USMAllocationError",
     "as_usm_memory",
+    "SyclIpcGetMemHandle",
+    "SyclIpcOpenMemHandle",
+    "SyclIpcCloseMemHandle",
 ]