Cmake同时编译cpp文件和cu文件

最新推荐文章于 2025-06-10 15:17:08 发布

原创

最新推荐文章于 2025-06-10 15:17:08 发布 · 1.5k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #cmake #cuda

本文介绍如何将C++源文件转换为CUDA可执行文件，重点在于如何使用`cuda_add_executable`替换`add_executable`，并在执行前设置CUDA源文件的特性。步骤包括设置编译选项和使用`set_source_files_properties`来指定CUDA源文件格式。

关键两步：
1.add_executable指令换为cuda_add_executable
2.在cuda_add_executable执行前添加set_source_files_properties(${主c++文件名} PROPERTIES CUDA_SOURCE_PROPERTY_FORMAT OBJ)

部分代码：

set(CUDA_GEN_CODE "-gencode=arch=compute_75,code=sm_75")
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS}  -std=c++11 -O0 -Xcompiler -fPIC -g -w ${CUDA_GEN_CODE}")
set_source_files_properties(yolov5.cpp PROPERTIES CUDA_SOURCE_PROPERTY_FORMAT OBJ)
cuda_add_executable(yolov5 ${PROJECT_SOURCE_DIR}/calibrator.cpp