Skip to content

Instantly share code, notes, and snippets.

@daidedou
Created June 5, 2020 15:32
Show Gist options
  • Save daidedou/dc0e43070195d3b4f8899eed3fc3062f to your computer and use it in GitHub Desktop.
Save daidedou/dc0e43070195d3b4f8899eed3fc3062f to your computer and use it in GitHub Desktop.
cmake_minimum_required(VERSION 3.7)
project(myproject)
#enable_language("CUDA")
find_package(CUDA QUIET)
# Check for CUDA ENV vars
IF(NOT DEFINED ENV{CUDA_PATH})
MESSAGE(FATAL_ERROR "CUDA_PATH Environment variable is not set.")
ENDIF(NOT DEFINED ENV{CUDA_PATH})
# Set the toolkit path
FILE(TO_CMAKE_PATH "$ENV{CUDA_PATH}" CUDA_TOOLKIT_ROOT_DIR)
SET(CUDA_TOOLKIT_ROOT_DIR ${CUDA_TOOLKIT_ROOT_DIR} CACHE STRING "Root directory of the Cuda Library" FORCE)
# Find the package
find_package(CUDA REQUIRED)
# Create and interface library as a link target (Requires CMake 3.7.0+)
add_library(cuda INTERFACE)
set_target_properties(cuda PROPERTIES
INTERFACE_INCLUDE_DIRECTORIES ${CUDA_INCLUDE_DIRS}
INTERFACE_LINK_LIBRARIES "${CUDA_LIBRARIES};${CUDA_CUFFT_LIBRARIES};${CUDA_CUBLAS_LIBRARIES}"
)
SET(CUDA_HOST_COMPILATION_CPP ON)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment