Merge pull request #7 from LucasGandel/cuda_include_dirs

COMP: Add CUDA include dirs to CudaCommon_INCLUDE_DIRS
RTKConsortium · Jan 17, 2022 · 2999fc3 · 2999fc3
2 parents deaedf2 + 8e2a7dc
commit 2999fc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -7,7 +7,7 @@ if(NOT ITK_SOURCE_DIR)
 endif()
 
 set(CudaCommon_LIBRARIES CudaCommon)
-set(CudaCommon_INCLUDE_DIRS ${CudaCommon_SOURCE_DIR}/include)
+set(CudaCommon_INCLUDE_DIRS ${CudaCommon_SOURCE_DIR}/include ${CUDAToolkit_INCLUDE_DIRS})
 
 # --------------------------------------------------------
 # Find ITK (required)

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -11,5 +11,5 @@ set(CudaCommon_Kernels
 
 itk_module_add_library(CudaCommon ${CudaCommon_SRCS} ${CudaCommon_Kernels})
 target_link_libraries(CudaCommon LINK_PUBLIC CUDA::cudart CUDA::cuda_driver)
-target_include_directories(CudaCommon PUBLIC ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
+target_include_directories(CudaCommon PUBLIC ${CUDAToolkit_INCLUDE_DIRS})
 set_property(TARGET CudaCommon PROPERTY CUDA_STANDARD ${CMAKE_CXX_STANDARD})