【CUDA】CMakeLists CUDAコンパイルを実現

2363 ワード

#   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

このうちcuhcuのヘッダファイルであり、cppcuhのファイルを参照すればCUDAのkernelを呼び出すことができる.