cmake/FindCUDA.cmake

   1 IF (NOT CUDA_HEADER_PREFIX)
   2   SET(CUDA_HEADER_PREFIX "/usr/local/cuda/include")
   3 ENDIF (NOT CUDA_HEADER_PREFIX)
   4
   5 FIND_PATH(CUDALIB_INCLUDE_DIR
   6   NAMES cuda.h
   7   PATHS ${CUDA_HEADER_PREFIX})
   8
   9 FIND_LIBRARY(CUDALIB_LIBRARY NAMES cuda)
  10
  11 IF (CUDALIB_INCLUDE_DIR)
  12   SET(CUDALIB_FOUND TRUE)
  13 ENDIF (CUDALIB_INCLUDE_DIR)
  14
  15 IF (CUDALIB_FOUND)
  16   IF (NOT CUDA_FIND_QUIETLY)
  17     MESSAGE(STATUS "Found CUDA: ${CUDALIB_INCLUDE_DIR}")
  18   ENDIF (NOT CUDA_FIND_QUIETLY)
  19 ELSE (CUDALIB_FOUND)
  20   IF (CUDA_FIND_REQUIRED)
  21     MESSAGE(FATAL_ERROR "Could not find CUDA")
  22   ENDIF (CUDA_FIND_REQUIRED)
  23 ENDIF (CUDALIB_FOUND)