cmake를 사용하여 cuda 파일을 cpp 파일에 올바르게 연결하는 방법

Question

cmake를 사용하여 .cpp 파일 및 .cu 파일을 포함하는 프로젝트를 컴파일하는 방법을 배우고 있습니다. 현재 main.cpp와 kernel.cu라는 두 개의 파일 만있는 장난감 예제로 작업하고 있습니다. 파일은

MAIN.CPP :

#include <stdlib.h> 
#include <string.h> 
extern void kernel_wrapper(int *a, int *b); 

int main(int argc, char *argv[]){ 
    int a = 2; 
    int b = 3; 

    printf("Input: a = %d, b = %d\n",a,b); 
    kernel_wrapper(&a, &b); 
    printf("Ran: a = %d, b = %d\n",a,b); 
    return 0; 
} 

kernel.cu 다음 cmake 파일이 게시물에서 영감을

//#include "cuPrintf.cu" 
#include <stdio.h> 
__global__ void kernel(int *a, int *b){ 
    int tx = threadIdx.x; 
// cuPrintf("tx = %d\n", tx); 
    switch(tx){ 
    case 0: 
     *a = *a + 10; 
     break; 
    case 1: 
     *b = *b + 3; 
     break; 
    default: 
     break; 
    } 
} 

void kernel_wrapper(int *a, int *b){ 
// cudaPrintfInit(); 
    //cuPrintf("Anything...?"); 
    printf("Anything...?\n"); 
    int *d_1, *d_2; 
    dim3 threads(2, 1); 
    dim3 blocks(1, 1); 

    cudaMalloc((void **)&d_1, sizeof(int)); 
    cudaMalloc((void **)&d_2, sizeof(int)); 

    cudaMemcpy(d_1, a, sizeof(int), cudaMemcpyHostToDevice); 
    cudaMemcpy(d_2, b, sizeof(int), cudaMemcpyHostToDevice); 

    kernel<<< blocks, threads >>>(d_1, d_2); 
    cudaMemcpy(a, d_1, sizeof(int), cudaMemcpyDeviceToHost); 
    cudaMemcpy(b, d_2, sizeof(int), cudaMemcpyDeviceToHost); 
    printf("Output: a = %d\n", a[0]); 
    cudaFree(d_1); 
    cudaFree(d_2); 

// cudaPrintfDisplay(stdout, true); 
// cudaPrintfEnd(); 
} 

: CMake script for CUDA 6.0 with C++11

set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -pthread -lpthread") 
cmake_minimum_required(VERSION 3.2) 
project(CUDAAndCP) 
find_package(CUDA REQUIRED) 

# For compilation ... 
# Specify target & source files to compile it from 

# Pass options to NVCC 
set(
    CUDA_NVCC_FLAGS 
    ${CUDA_NVCC_FLAGS}; 
    -O3 -gencode arch=compute_30,code=sm_30; 
    --std=c++11 
    ) 

cuda_add_library(kernel_obj kernel.cu) 
cuda_add_executable(main main.cpp) 
target_link_libraries(main ${CUDA_LIBRARIES}) 
target_link_libraries(main kernel_obj) 

내가받을 수 있나요 컴파일 된 메인 바이너리 파일이지만 실행할 때 결과는

입니다. 대신

Input: a = 2, b = 3 
Anything...? 
Output: a = 2 
Ran: a = 12, b = 6 

의 5,

Input: a = 2, b = 3 
Anything...? 
Output: a = 2 
Ran: a = 2, b = 3 

나는 그래서는 CUDA 파일이 기본 바이너리에 제대로 연결되지 않은 것 같다

g++ -c main.cpp 
nvcc -c kernel.cu 
nvcc -o main main.o kernel.o 

을 실행하여 올바른 기본 바이너리 파일을 얻었다. 나는 왜 그리고 어떤 도움이 감사인지 정말로 이해하지 않는다!

우분투 14.04, 쿠다 7.5 및 cmake 3.2.0에서 실행 중입니다.

Answer 1

코드 생성에있어 까다로운 부분이 있다고 생각합니다. 내 플랫폼은 귀하의 것과 동일하며 GTX 980을 사용하고 있습니다.

실제 문제는 arch=compute_30,code=sm_30;입니다. 이 값을 52로 변경하면 모든 것이 작동합니다.

기본 명령 줄을 사용하여 빌드 한 경우 compute_20 및 sm_20 코드를 생성하려면 CUDA 7.5 nvcc이 기본값입니다.