[SYCL][CUDA][PI] Added CUDA-specific pi_mem_advice values

Author: t4c1

sycl/include/CL/sycl/detail/pi.h

@@ -440,7 +440,17 @@ typedef enum {
 
 typedef enum {
   // Device-specific value opaque in PI API.
-  PI_MEM_ADVISE_UNKNOWN
+  PI_MEM_ADVISE_UNKNOWN,
+  PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY = 1,
+  PI_MEM_ADVISE_CUDA_UNSET_READ_MOSTLY = 2,
+  PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION = 3,
+  PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION = 4,
+  PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY = 5,
+  PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY = 6,
+  PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST = 7,
+  PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION_HOST = 8,
+  PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY_HOST = 9,
+  PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY_HOST = 10,
 } _pi_mem_advice;
 
 typedef enum {

sycl/plugins/cuda/pi_cuda.cpp

@@ -4709,9 +4709,30 @@ pi_result cuda_piextUSMEnqueueMemAdvise(pi_queue queue, const void *ptr,
       event_ptr->start();
     }
 
-    result = PI_CHECK_ERROR(
-        cuMemAdvise((CUdeviceptr)ptr, length, (CUmem_advise)advice,
-                    queue->get_context()->get_device()->get()));
+    switch(advice){
+      case PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY:
+      case PI_MEM_ADVISE_CUDA_UNSET_READ_MOSTLY:
+      case PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION:
+      case PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION:
+      case PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY:
+      case PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY:
+        result = PI_CHECK_ERROR(
+            cuMemAdvise((CUdeviceptr)ptr, length, (CUmem_advise)advice,
+                        queue->get_context()->get_device()->get()));
+        break;
+      case PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST:
+      case PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION_HOST:
+      case PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY_HOST:
+      case PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY_HOST:
+        result = PI_CHECK_ERROR(
+            cuMemAdvise((CUdeviceptr)ptr, length, (CUmem_advise)(advice - 
+                (PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST - 
+                  PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION)),
+                        -1));
+        break;
+      default:
+        cl::sycl::detail::pi::die("Unknown advice");
+    }
     if (event) {
       result = event_ptr->record();
       *event = event_ptr.release();