Skip to content

Commit 29a3503

Browse files
committed
addressed review comments
1 parent 2a8b13d commit 29a3503

File tree

2 files changed

+16
-13
lines changed

2 files changed

+16
-13
lines changed

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

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -441,16 +441,16 @@ typedef enum {
441441
typedef enum {
442442
// Device-specific value opaque in PI API.
443443
PI_MEM_ADVISE_UNKNOWN,
444-
PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY = 1,
445-
PI_MEM_ADVISE_CUDA_UNSET_READ_MOSTLY = 2,
446-
PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION = 3,
447-
PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION = 4,
448-
PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY = 5,
449-
PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY = 6,
450-
PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST = 7,
451-
PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION_HOST = 8,
452-
PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY_HOST = 9,
453-
PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY_HOST = 10,
444+
PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY = 101,
445+
PI_MEM_ADVISE_CUDA_UNSET_READ_MOSTLY = 102,
446+
PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION = 103,
447+
PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION = 104,
448+
PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY = 105,
449+
PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY = 106,
450+
PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST = 107,
451+
PI_MEM_ADVISE_CUDA_UNSET_PREFERRED_LOCATION_HOST = 108,
452+
PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY_HOST = 109,
453+
PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY_HOST = 110,
454454
} _pi_mem_advice;
455455

456456
typedef enum {

sycl/plugins/cuda/pi_cuda.cpp

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4717,7 +4717,10 @@ pi_result cuda_piextUSMEnqueueMemAdvise(pi_queue queue, const void *ptr,
47174717
case PI_MEM_ADVISE_CUDA_SET_ACCESSED_BY:
47184718
case PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY:
47194719
result = PI_CHECK_ERROR(
4720-
cuMemAdvise((CUdeviceptr)ptr, length, (CUmem_advise)advice,
4720+
cuMemAdvise((CUdeviceptr)ptr,
4721+
length,
4722+
(CUmem_advise)(
4723+
advice - PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY + 1),
47214724
queue->get_context()->get_device()->get()));
47224725
break;
47234726
case PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST:
@@ -4726,10 +4729,10 @@ pi_result cuda_piextUSMEnqueueMemAdvise(pi_queue queue, const void *ptr,
47264729
case PI_MEM_ADVISE_CUDA_UNSET_ACCESSED_BY_HOST:
47274730
result = PI_CHECK_ERROR(cuMemAdvise(
47284731
(CUdeviceptr)ptr, length,
4729-
(CUmem_advise)(advice -
4732+
(CUmem_advise)(advice - PI_MEM_ADVISE_CUDA_SET_READ_MOSTLY + 1 -
47304733
(PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION_HOST -
47314734
PI_MEM_ADVISE_CUDA_SET_PREFERRED_LOCATION)),
4732-
-1));
4735+
CU_DEVICE_CPU));
47334736
break;
47344737
default:
47354738
cl::sycl::detail::pi::die("Unknown advice");

0 commit comments

Comments
 (0)