如何编写支持C++与CUDA编译器的条件Makefile？

一、宏封装方案优化 #

当前写法存在隐患：若编译时未定义WITH_CUDA，但运行时with_cuda被设为true，会进入空分支导致无操作。以下是更可靠的优化方案：

1. 编译时绑定CUDA支持状态 #

在公共头文件中定义编译时常量，限制运行时with_cuda的合法取值：

#ifdef WITH_CUDA
constexpr bool HAS_CUDA = true;
#else
constexpr bool HAS_CUDA = false;
#endif

初始化with_cuda时添加断言，直接拦截无效状态：

#include <cassert>

int main() {
    bool with_cuda = false;
    // 从命令行/配置读取是否启用CUDA
    if (/* 用户请求启用CUDA */) {
        // 编译时无CUDA支持则直接终止程序
        assert(HAS_CUDA && "当前程序版本不支持CUDA");
        with_cuda = true;
    }

    if (with_cuda) {
#ifdef WITH_CUDA
        call_kernel(parameters);
#else
        // 兜底处理（理论上不会执行到此处）
        call_cpu_function(parameters);
#endif
    } else {
        call_cpu_function(parameters);
    }
}

2. 编译时替换实现（无需运行时分支） #

如果不需要运行时动态切换CUDA/CPU实现，可直接用宏统一函数调用，彻底消除分支隐患：

#ifdef WITH_CUDA
#define RUN_OPERATION(params) call_kernel(params)
#else
#define RUN_OPERATION(params) call_cpu_function(params)
#endif

// 代码中直接调用统一接口
RUN_OPERATION(parameters);

二、自动检测CUDA的构建方案 #

1. 原生Makefile实现 #

通过shell命令检测nvcc和CUDA头文件路径，自动切换编译器与编译选项：

# 检测CUDA工具包
NVCC := $(shell which nvcc 2>/dev/null)
CUDA_INC := $(shell [ -d /usr/local/cuda/include ] && echo /usr/local/cuda/include)

# 默认配置
CC := g++
CFLAGS := -Wall -O2
LDFLAGS :=
SRC := files.cpp
OBJ := $(SRC:.cpp=.o)
TARGET := my_program

# 检测到CUDA时更新配置
ifneq ($(NVCC),)
    CC := nvcc
    CFLAGS += -DWITH_CUDA -I$(CUDA_INC)
    SRC += other_files.cu
    OBJ += other_files.o
    LDFLAGS += -lcudart
endif

all: $(TARGET)

$(TARGET): $(OBJ)
    $(CC) $(OBJ) -o $@ $(LDFLAGS)

%.o: %.cpp
    $(CC) $(CFLAGS) -c $< -o $@

%.o: %.cu
    $(NVCC) $(CFLAGS) -c $< -o $@

clean:
    rm -f $(OBJ) $(TARGET)

2. CMake实现（推荐跨平台） #

CMake自带CUDA检测模块，无需手动处理路径与编译器切换，跨平台兼容性更好：

cmake_minimum_required(VERSION 3.18)
project(my_cuda_program)

# 设置C++标准
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# 静默查找CUDA工具包
find_package(CUDAToolkit QUIET)

# 基础源文件
set(SRC_FILES files.cpp)

if(CUDAToolkit_FOUND)
    # 添加CUDA源文件
    list(APPEND SRC_FILES other_files.cu)
    # 定义WITH_CUDA宏
    add_definitions(-DWITH_CUDA)
    # 链接CUDA运行时库
    link_libraries(CUDAToolkit::cudart)
    # 启用CUDA编译支持
    enable_language(CUDA)
endif()

# 生成可执行文件
add_executable(my_program ${SRC_FILES})

使用方式：

mkdir build && cd build
cmake ..
make

内容的提问来源于stack exchange，提问作者MyUserIsThis