NVIDIA/CCCL项目中Thrust库的CMake配置最佳实践

概述

在NVIDIA/CCCL项目中，Thrust作为并行算法库提供了多种后端实现方式，包括CPP、OMP、TBB和CUDA。本文将详细介绍如何通过CMake正确配置Thrust库的不同后端实现，解决实际开发中遇到的常见问题。

CMake配置基础

基本配置方法

要使用CCCL中的Thrust库，最基本的CMake配置如下：

cmake_minimum_required(VERSION 3.26)
project(thrust-test VERSION 0.1 LANGUAGES CXX CUDA)
find_package(CCCL 2.7 REQUIRED)
add_executable(main main.cu)
target_link_libraries(main PRIVATE CCCL::CCCL)

这种配置默认使用CUDA后端。要切换后端，可以在配置时通过命令行参数指定：

cmake -B build . -DCCCL_THRUST_DEVICE_SYSTEM=OMP

后端选项详解

CCCL_THRUST_DEVICE_SYSTEM支持以下选项：

• CPP：使用纯C++实现
• OMP：使用OpenMP并行
• TBB：使用Intel TBB并行
• CUDA：使用CUDA加速

这些选项在CMake GUI中默认被标记为"高级"，需要启用高级模式才能看到（在ccmake中按t键，在cmake-gui中勾选Advanced选项）。

高级配置技巧

条件性启用CUDA语言

对于不需要CUDA后端的场景，可以避免加载CUDA语言支持：

project(thrust-test VERSION 0.1 LANGUAGES CXX) # 始终启用CXX

if(CCCL_THRUST_DEVICE_SYSTEM STREQUAL CUDA)
  enable_language(CUDA) # 仅在需要时启用CUDA
endif()

find_package(CCCL 2.7 REQUIRED)

多后端同时配置

如果需要在一个项目中同时使用多个后端，可以使用thrust_create_target函数：

find_package(CCCL 2.7 REQUIRED)

# 创建使用不同后端的Thrust目标
thrust_create_target(ThrustCPP DEVICE CPP)
thrust_create_target(ThrustOMP DEVICE OMP)
thrust_create_target(ThrustTBB DEVICE TBB)

# 为不同后端创建不同的可执行文件
add_executable(main_cpp main.cpp)
target_link_libraries(main_cpp PRIVATE ThrustCPP)

add_executable(main_omp main.cpp)
target_link_libraries(main_omp PRIVATE ThrustOMP)

add_executable(main_tbb main.cpp)
target_link_libraries(main_tbb PRIVATE ThrustTBB)

处理文件扩展名问题

当使用CUDA后端时，通常需要将源文件命名为.cu扩展名。如果希望保持.cpp扩展名，可以设置源文件的语言属性：

set_source_files_properties(main.cpp PROPERTIES LANGUAGE CUDA)

代码中检测后端类型

在源代码中，仍然可以使用传统的THRUST_DEVICE_SYSTEM宏来检测当前使用的后端：

#include <iostream>
#include <thrust/device_vector.h>

int main()
{
#if THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_CPP
    std::cout << "使用CPP后端";
#elif THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_OMP
    std::cout << "使用OpenMP后端";
#elif THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_CUDA
    std::cout << "使用CUDA后端";
#endif
    return 0;
}

最佳实践总结

1. 优先使用CCCL::CCCL或CCCL::Thrust目标，而非直接使用Thrust包
2. 对于简单项目，使用CCCL_THRUST_DEVICE_SYSTEM变量切换后端
3. 对于复杂项目，考虑使用thrust_create_target创建多个目标
4. 合理处理CUDA语言的条件性加载
5. 注意源文件扩展名和语言属性的设置
6. 在代码中使用标准宏检测后端类型

通过遵循这些实践，可以更高效地在项目中使用Thrust库的不同后端实现，同时保持构建系统的清晰和可维护性。