COMP: provide nvcc location to CMake package CUDAToolkit

nvcc is found by find_language(CUDA) but the language is not immediately set to let cmake check CMake and CUDA version
SimonRit · Oct 14, 2021 · 2a05f38 · 2a05f38
1 parent b9ff8dc
commit 2a05f38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/itk-module-init.cmake b/itk-module-init.cmake
@@ -9,6 +9,7 @@ check_language(CUDA)
 # Determine default value for RTK_USE_CUDA
 set(RTK_USE_CUDA_DEFAULT OFF)
 if (CMAKE_CUDA_COMPILER)
+  set(CUDAToolkit_NVCC_EXECUTABLE ${CMAKE_CUDA_COMPILER})
   find_package(CUDAToolkit)
   if(NOT CUDAToolkit_FOUND)
     message(WARNING "CUDAToolkit not found (available since CMake v3.17). RTK_USE_CUDA set to OFF.")