From 8e2a7dc677976a4e055e0e8f4148f21df7150613 Mon Sep 17 00:00:00 2001 From: LucasGandel Date: Mon, 17 Jan 2022 13:55:12 +0100 Subject: [PATCH] COMP: Add CUDA include dirs to CudaCommon_INCLUDE_DIRS Fixes "error: cuda.h not found" when wrapping for python. Prefer CUDAToolkit_INCLUDE_DIRS over CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES when using find_package(CUDAToolkit). --- CMakeLists.txt | 2 +- src/CMakeLists.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 434898c..a513ed7 100644 --- 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 index dac797c..5facb4c 100644 --- 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})