17 lines
474 B
CMake
17 lines
474 B
CMake
|
set(DEVICE_COMPILER "nvcc")
|
||
|
|
set(CUDA_COMPILER "${DEVICE_COMPILER}" CACHE STRING "Select the device compiler")
|
||
|
|
|
||
|
|
if (CMAKE_CXX_COMPILER_ID STREQUAL "Clang")
|
||
|
|
list(APPEND DEVICE_COMPILER "clang")
|
||
|
|
endif()
|
||
|
|
|
||
|
|
set_property(CACHE CUDA_COMPILER PROPERTY STRINGS "${DEVICE_COMPILER}")
|
||
|
|
|
||
|
|
list(APPEND CMAKE_PREFIX_PATH "$ENV{CUDA_ROOT}")
|
||
|
|
list(APPEND CMAKE_PREFIX_PATH "$ENV{CMAKE_PREFIX_PATH}")
|
||
|
|
|
||
|
|
set(CUDA_STATIC ON)
|
||
|
|
find_package(CUDA 8.0 REQUIRED)
|
||
|
|
|
||
|
|
set(LIBS ${LIBS} ${CUDA_LIBRARIES})
|