Implemented SyclQueue.prefetch, SyclQueue.mem_advise

Author: oleksandr-pavlyk

dpctl/_backend.pxd

@@ -177,6 +177,10 @@ cdef extern from "dppl_sycl_queue_interface.h":
     cdef void DPPLQueue_Wait (const DPPLSyclQueueRef QRef)
     cdef void DPPLQueue_Memcpy (const DPPLSyclQueueRef Q,
                                 void *Dest, const void *Src, size_t Count)
+    cdef void DPPLQueue_Prefetch (const DPPLSyclQueueRef Q,
+                                const void *Src, size_t Count)
+    cdef void DPPLQueue_MemAdvise (const DPPLSyclQueueRef Q,
+                                   const void *Src, size_t Count, int Advice)
 
 
 cdef extern from "dppl_sycl_queue_manager.h":

dpctl/_sycl_core.pxd

@@ -117,7 +117,9 @@ cdef class SyclQueue:
                             list lS=*, list dEvents=*)
     cpdef void wait (self)
     cdef DPPLSyclQueueRef get_queue_ref (self)
-    cpdef memcpy (self, dest, src, int count)
+    cpdef memcpy (self, dest, src, size_t count)
+    cpdef prefetch (self, ptr, size_t count=*)
+    cpdef mem_advise (self, ptr, size_t count, int mem)
 
 
 cpdef SyclQueue get_current_queue()

dpctl/_sycl_core.pyx

@@ -541,22 +541,48 @@ cdef class SyclQueue:
     cpdef void wait (self):
         DPPLQueue_Wait(self._queue_ref)
 
-    cpdef memcpy (self, dest, src, int count):
+    cpdef memcpy (self, dest, src, size_t count):
         cdef void *c_dest
         cdef void *c_src
 
         if isinstance(dest, Memory):
             c_dest = <void*>(<Memory>dest).memory_ptr
         else:
-            raise TypeError("Parameter dest should be Memory.")
+            raise TypeError("Parameter dest should have type Memory.")
 
         if isinstance(src, Memory):
             c_src = <void*>(<Memory>src).memory_ptr
         else:
-            raise TypeError("Parameter src should be Memory.")
+            raise TypeError("Parameter src should have type Memory.")
 
         DPPLQueue_Memcpy(self._queue_ref, c_dest, c_src, count)
 
+    cpdef prefetch (self, mem, size_t count=0):
+       cdef void *ptr
+
+       if isinstance(mem, Memory):
+           ptr = <void*>(<Memory>mem).memory_ptr
+       else:
+           raise TypeError("Parameter mem should have type Memory")
+
+       if (count <=0 or count > self.nbytes):
+           count = self.nbytes
+
+       DPPLQueue_Prefetch(self._queue_ref, ptr, count)
+
+    cpdef mem_advise (self, mem, size_t count, int advice):
+       cdef void *ptr
+
+       if isinstance(mem, Memory):
+           ptr = <void*>(<Memory>mem).memory_ptr
+       else:
+           raise TypeError("Parameter mem should have type Memory")
+
+       if (count <=0 or count > self.nbytes):
+           count = self.nbytes
+
+       DPPLQueue_MemAdvise(self._queue_ref, ptr, count, advice)
+
 
 cdef class _SyclRTManager:
     ''' Wrapper for the C API's sycl queue manager interface.

dpctl/tests/test_sycl_queue_memcpy.py

@@ -61,13 +61,13 @@ def test_memcpy_type_error(self):
             q.memcpy(None, mobj, 3)
 
         self.assertEqual(type(cm.exception), TypeError)
-        self.assertEqual(str(cm.exception), "Parameter dest should be Memory.")
+        self.assertEqual(str(cm.exception), "Parameter dest should have type Memory.")
 
         with self.assertRaises(TypeError) as cm:
             q.memcpy(mobj, None, 3)
 
         self.assertEqual(type(cm.exception), TypeError)
-        self.assertEqual(str(cm.exception), "Parameter src should be Memory.")
+        self.assertEqual(str(cm.exception), "Parameter src should have type Memory.")
 
 
 if __name__ == "__main__":