Added DPPLaligned_alloc_* functions to backend

Author: oleksandr-pavlyk

backends/include/dppl_sycl_usm_interface.h

@@ -42,6 +42,16 @@ DPPL_API
 __dppl_give DPPLSyclUSMRef
 DPPLmalloc_shared (size_t size, __dppl_keep const DPPLSyclQueueRef QRef);
 
+/*!
+ * @brief Crete USM shared memory.
+ *
+ * @return The pointer to USM shared memory with requested alignment.
+ */
+DPPL_API
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_shared (size_t alignment, size_t size,
+			  __dppl_keep const DPPLSyclQueueRef QRef);
+
 /*!
  * @brief Crete USM host memory.
  *
@@ -51,6 +61,16 @@ DPPL_API
 __dppl_give DPPLSyclUSMRef
 DPPLmalloc_host (size_t size, __dppl_keep const DPPLSyclQueueRef QRef);
 
+/*!
+ * @brief Crete USM host memory.
+ *
+ * @return The pointer to USM host memory with requested alignment.
+ */
+DPPL_API
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_host (size_t alignment, size_t size,
+			__dppl_keep const DPPLSyclQueueRef QRef);
+
 /*!
  * @brief Crete USM device memory.
  *
@@ -60,6 +80,16 @@ DPPL_API
 __dppl_give DPPLSyclUSMRef
 DPPLmalloc_device (size_t size, __dppl_keep const DPPLSyclQueueRef QRef);
 
+/*!
+ * @brief Crete USM device memory.
+ *
+ * @return The pointer to USM device memory with requested alignment.
+ */
+DPPL_API
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_device (size_t alignment, size_t size,
+			  __dppl_keep const DPPLSyclQueueRef QRef);
+
 /*!
  * @brief Free USM memory.
  *

backends/source/dppl_sycl_usm_interface.cpp

@@ -48,6 +48,15 @@ DPPLmalloc_shared (size_t size, __dppl_keep const DPPLSyclQueueRef QRef)
     return wrap(Ptr);
 }
 
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_shared (size_t alignment, size_t size,
+			  __dppl_keep const DPPLSyclQueueRef QRef)
+{
+    auto Q = unwrap(QRef);
+    auto Ptr = aligned_alloc_shared(alignment, size, *Q);
+    return wrap(Ptr);
+}
+
 __dppl_give DPPLSyclUSMRef
 DPPLmalloc_host (size_t size, __dppl_keep const DPPLSyclQueueRef QRef)
 {
@@ -56,6 +65,15 @@ DPPLmalloc_host (size_t size, __dppl_keep const DPPLSyclQueueRef QRef)
     return wrap(Ptr);
 }
 
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_host (size_t alignment, size_t size,
+			__dppl_keep const DPPLSyclQueueRef QRef)
+{
+    auto Q = unwrap(QRef);
+    auto Ptr = aligned_alloc_host(alignment, size, *Q);
+    return wrap(Ptr);
+}
+
 __dppl_give DPPLSyclUSMRef
 DPPLmalloc_device (size_t size, __dppl_keep const DPPLSyclQueueRef QRef)
 {
@@ -64,6 +82,15 @@ DPPLmalloc_device (size_t size, __dppl_keep const DPPLSyclQueueRef QRef)
     return wrap(Ptr);
 }
 
+__dppl_give DPPLSyclUSMRef
+DPPLaligned_alloc_device (size_t alignment, size_t size,
+			  __dppl_keep const DPPLSyclQueueRef QRef)
+{
+    auto Q = unwrap(QRef);
+    auto Ptr = aligned_alloc_device(alignment, size, *Q);
+    return wrap(Ptr);
+}
+
 void DPPLfree_with_queue (__dppl_take DPPLSyclUSMRef MRef,
                           __dppl_keep const DPPLSyclQueueRef QRef)
 {

dpctl/_backend.pxd

@@ -203,6 +203,12 @@ cdef extern from "dppl_sycl_usm_interface.h":
     cdef DPPLSyclUSMRef DPPLmalloc_shared (size_t size, DPPLSyclQueueRef QRef)
     cdef DPPLSyclUSMRef DPPLmalloc_host (size_t size, DPPLSyclQueueRef QRef)
     cdef DPPLSyclUSMRef DPPLmalloc_device (size_t size, DPPLSyclQueueRef QRef)
+    cdef DPPLSyclUSMRef DPPLaligned_alloc_shared (size_t alignment,
+                                                  size_t size, DPPLSyclQueueRef QRef)
+    cdef DPPLSyclUSMRef DPPLaligned_alloc_host (size_t alignment,
+                                                size_t size, DPPLSyclQueueRef QRef)
+    cdef DPPLSyclUSMRef DPPLaligned_alloc_device (size_t alignment,
+                                                  size_t size, DPPLSyclQueueRef QRef)
     cdef void DPPLfree_with_queue (DPPLSyclUSMRef MRef,
                                    DPPLSyclQueueRef QRef)
     cdef void DPPLfree_with_context (DPPLSyclUSMRef MRef,