File tree Expand file tree Collapse file tree 2 files changed +10
-7
lines changed Expand file tree Collapse file tree 2 files changed +10
-7
lines changed Original file line number Diff line number Diff line change @@ -4,22 +4,24 @@ message(STATUS "Including the PI API CUDA backend.")
4
4
# we only require the CUDA driver API to be used
5
5
# CUDA_CUDA_LIBRARY variable defines the path to libcuda.so, the CUDA Driver API library.
6
6
7
- find_package (CUDA 10.1 REQUIRED )
7
+ find_package (CUDAToolkit 10.1 REQUIRED )
8
+
9
+ message (STATUS "Including the PI API CUDA backend." )
8
10
9
11
# Make imported library global to use it within the project.
10
12
add_library (cudadrv SHARED IMPORTED GLOBAL )
11
13
12
14
if (WIN32 )
13
15
set_target_properties (
14
16
cudadrv PROPERTIES
15
- IMPORTED_IMPLIB ${CUDA_CUDA_LIBRARY }
16
- INTERFACE_INCLUDE_DIRECTORIES ${CUDA_INCLUDE_DIRS }
17
+ IMPORTED_IMPLIB ${CUDA_cuda_driver_LIBRARY }
18
+ INTERFACE_INCLUDE_DIRECTORIES ${CUDAToolkit_INCLUDE_DIRS }
17
19
)
18
20
else ()
19
21
set_target_properties (
20
22
cudadrv PROPERTIES
21
- IMPORTED_LOCATION ${CUDA_CUDA_LIBRARY }
22
- INTERFACE_INCLUDE_DIRECTORIES ${CUDA_INCLUDE_DIRS }
23
+ IMPORTED_LOCATION ${CUDA_cuda_driver_LIBRARY }
24
+ INTERFACE_INCLUDE_DIRECTORIES ${CUDAToolkit_INCLUDE_DIRS }
23
25
)
24
26
endif ()
25
27
Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ add_dependencies(sycl_pi_trace_collector pi-pretty-printers ze-pretty-printers)
60
60
61
61
if (SYCL_BUILD_PI_CUDA )
62
62
63
- find_package (CUDA 10.1 REQUIRED )
63
+ find_package (CUDAToolkit 10.1 REQUIRED )
64
64
target_include_directories (sycl_pi_trace_collector
65
65
PRIVATE
66
66
"${CUDA_TOOLKIT_ROOT_DIR} /extras/CUPTI/include"
@@ -75,7 +75,8 @@ if(SYCL_BUILD_PI_CUDA)
75
75
add_custom_target (cuda-pretty-printers
76
76
COMMAND ${Python3_EXECUTABLE}
77
77
${CMAKE_CURRENT_SOURCE_DIR} /generate_cuda_pretty_printers.py
78
- ${CUDA_TOOLKIT_ROOT_DIR} /extras/CUPTI/include/generated_cuda_meta.h
78
+ # ${CUDA_TOOLKIT_ROOT_DIR}/extras/CUPTI/include/generated_cuda_meta.h
79
+ /usr/local/cuda/targets/x86_64-linux/include/generated_cuda_meta.h
79
80
DEPENDS pi_cuda
80
81
BYPRODUCTS
81
82
${CMAKE_CURRENT_BINARY_DIR} /cuda_printers.def
You can’t perform that action at this time.
0 commit comments