CMakeLists.txt

cmake_minimum_required(VERSION 3.16)
# 导出编译命令
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

# 指定cuda编译路径
set(CMAKE_CUDA_COMPILER "/usr/local/cuda-12.1/bin/nvcc")

# 使用 C++ 和 CUDA 语言
project(cuda_learn CXX CUDA)

include(cmake/cuda.cmake)
# C++ 标准为 C++17，CUDA 标准为 14。
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)
set(CMAKE_CUDA_STANDARD 14)
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_SOURCE_DIR}/lib)

find_package(GTest REQUIRED)
find_package(glog REQUIRED)

aux_source_directory(project/src/ DIR_MAIN)
aux_source_directory(project/src/operators/gpu/ DIR_KERNEL_CUDA)
aux_source_directory(project/src/operators/ DIR_OPERATORS)
aux_source_directory(project/src/base/ DIR_BASE)

add_library(cudalearn SHARED ${DIR_KERNEL_CUDA} ${DIR_BASE} ${DIR_MAIN} ${DIR_OPERATORS})

target_link_libraries(cudalearn glog::glog gtest gtest_main pthread cudart)
target_link_directories(cudalearn PUBLIC ${CMAKE_CUDA_COMPILER_LIBRARY_ROOT}/lib64)

target_include_directories(cudalearn PUBLIC ${glog_INCLUDE_DIR})
target_include_directories(cudalearn PUBLIC ${PROJECT_SOURCE_DIR}/project/include)
target_include_directories(cudalearn PUBLIC ${PROJECT_SOURCE_DIR}/project/include/base)
target_include_directories(cudalearn PUBLIC ${PROJECT_SOURCE_DIR}/project/src/operators)
target_include_directories(cudalearn PUBLIC ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})

# 启用了 CUDA 的可分离编译功能（Separable Compilation），以便分模块编译 CUDA 代码：
set_target_properties(cudalearn PROPERTIES CUDA_SEPARABLE_COMPILATION ON)

add_subdirectory(test)

# -----------------------以下是编译主程序--------------------------
set(link_ext_lib glog::glog GTest::gtest)

add_executable(demo project/src/main.cpp)

target_include_directories(demo PUBLIC ${glog_INCLUDE_DIR})
target_include_directories(demo PUBLIC ${GTest_INCLUDE_DIR})
# 这一行可能有点重复引用
target_include_directories(demo PUBLIC ${PROJECT_SOURCE_DIR}/project/include)

target_link_directories(demo PUBLIC ${PROJECT_SOURCE_DIR}/lib)

target_link_libraries(demo ${link_ext_lib})
target_link_libraries(demo cudalearn)

set_target_properties(demo PROPERTIES WORKING_DIRECTORY ${CMAKE_SOURCE_DIR})
set_target_properties(demo PROPERTIES CUDA_SEPARABLE_COMPILATION ON)