Fix bug in script for current CMakes

CLIUtils · Apr 12, 2018 · 2acdd6d · 2acdd6d
1 parent 4e52e4d
commit 2acdd6d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/select_compute_arch.cmake b/select_compute_arch.cmake
@@ -23,7 +23,11 @@ set(CUDA_KNOWN_GPU_ARCHITECTURES  "Fermi" "Kepler" "Maxwell")
 # This list will be used for CUDA_ARCH_NAME = Common option (enabled by default)
 set(CUDA_COMMON_GPU_ARCHITECTURES "3.0" "3.5" "5.0")
 
-if(CMAKE_CUDA_COMPILER_LOADED) # CUDA as a language
+if(NOT CMAKE_CUDA_COMPILER_LOADED AND DEFINED CMAKE_CUDA_COMPILER_ID)
+    set(CMAKE_CUDA_COMPILER_LOADED 1)
+endif()
+
+if(CMAKE_CUDA_COMPILER_LOADED OR DEFINED CMAKE_CUDA_COMPILER_ID) # CUDA as a language
   if(CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA")
     set(CUDA_VERSION "${CMAKE_CUDA_COMPILER_VERSION}")
   endif()