使用cmake和3.5计算功能编译CUDA代码

Posted 2021-04-06

我需要编译一个使用cmake动态并行的CUDA代码。代码是：

#include <stdio.h>

__global__ void childKernel() {
    printf("Hello ");    
}

__global__ void parentKernel() {    
    childKernel<<<1,1>>>();
    cudaDeviceSynchronize();    
    printf("World!
");    
}

int main(int argc, char **argv){

    parentKernel<<<1, 1>>>();

    return 0;
}

并且cmake如下：

cmake_minimum_required(VERSION 2.8)

find_package(CUDA QUIET REQUIRED)

include_directories(/usr/include)
include_directories(/usr/local/cuda/lib)
include_directories(/usr/local/cuda-8.0/lib)
include_directories(/usr/local/cuda/include)
include_directories(/usr/local/cuda-8.0/include)

set(CUDA_SEPARABLE_COMPILATION ON)
find_package(CUDA QUIET REQUIRED)
set(CUDA_PROPAGATE_HOST_FLAGS OFF)

set(
    CUDA_NVCC_FLAGS 
    ${CUDA_NVCC_FLAGS}; 
    -arch=compute_35 -rdc=true -lcudadevrt
)

cuda_add_executable(
    prova 
    test.cu
)

我试图用nvcc直接编译代码传递-arch=compute_35 -rdc=true -lcudadevrt并且编译完美，但是当我尝试使用cmake编译它时会返回以下错误：

CMakeFiles/prova.dir/prova_intermediate_link.o: In function `__cudaRegisterLinkedBinary_66_tmpxft_00001101_00000000_13_cuda_device_runtime_compute_62_cpp1_ii_8b1a5d37':
link.stub:(.text+0xcc): undefined reference to `__fatbinwrap_66_tmpxft_00001101_00000000_13_cuda_device_runtime_compute_62_cpp1_ii_8b1a5d37'
link.stub:(.text+0xd0): undefined reference to `__fatbinwrap_66_tmpxft_00001101_00000000_13_cuda_device_runtime_compute_62_cpp1_ii_8b1a5d37'
collect2: error: ld returned 1 exit status
CMakeFiles/prova.dir/build.make:200: recipe for target 'prova' failed
make[2]: *** [prova] Error 1
CMakeFiles/Makefile2:67: recipe for target 'CMakeFiles/prova.dir/all' failed
make[1]: *** [CMakeFiles/prova.dir/all] Error 2
Makefile:83: recipe for target 'all' failed
make: *** [all] Error 2

答案

您看到的未定义符号在libcudadevrt.a中定义。我看到你告诉CMake链接到它，但似乎编译器找不到它。在命令行上尝试：

VERBOSE=1 make

并检查输出，看看你是否正在搜索/usr/local/cuda/lib64库。

这可能是无关的，但我也注意到你正在指示CMake搜索/usr/local/cuda/lib和/usr/local/cuda-8.0/lib的头文件。这是不正确的，因为这些目录只包含库。

另一答案

谢谢你的回答，我试图使用Tegra X2（compute_62）的正确计算能力，我已经检查了详细make的输出，我发现了以下输出：

CMakeFiles/prova.dir/prova_generated_test.cu.o CMakeFiles/prova.dir/prova_intermediate_link.o  -o prova -rdynamic /usr/local/cuda-8.0/lib64/libcudart_static.a -lpthread -ldl -lrt

似乎链接器试图搜索/usr/local/cuda-8.0/lib64/libcudart_static.a而不是libcudadevrt.a