# 设置CMake版本
cmake_minimum_required(VERSION 3.5)
project(Test1)
set(EXECUTABLE_OUTPUT_PATH ../)
# 找到CUDA路径并导入头文件
find_package(CUDA REQUIRED)
include_directories(${CUDA_INCLUDE_DIRS})
# 设置C++版本
set(CMAKE_CXX_STANDARD 11)
# 检索出源文件与头文件
file(GLOB_RECURSE TEST_HEAD ${CMAKE_CURRENT_LIST_DIR}/src/.h ${CMAKE_CURRENT_LIST_DIR}/src/.cuh)
file(GLOB TEST_SRC ${CMAKE_CURRENT_LIST_DIR}/src/*.cpp ${CMAKE_CURRENT_LIST_DIR}/src/*.cu)
# 设置CUAD编译配置
set(CUDA_NVCC_FLAGS "-g -G")
# build option
set(GENCODE -gencode=arch=compute_35,code=sm_35)
set(GENCODE ${GENCODE} -gencode=arch=compute_30,code=sm_30)
set(GENCODE ${GENCODE} -gencode=arch=compute_20,code=sm_20)
set(GENCODE ${GENCODE} -gencode=arch=compute_10,code=sm_10)
# 生成可执行文件
cuda_add_executable(Test ${TEST_SRC} ${TEST_HEAD} OPTIONS ${GENCODE})
文档结构
Test/
├── build
├── CMakeLists.txt
├── data
├── README.md
└── src
├── func.cu
├── func.cuh
└── main.cpp
其中cuh
是cu
的头文件,在cpp
中引用cuh
文件就可以调用CUDA的kernel 。