如何用Go语言的cgo编译Cuda源码？

我用 cuda-c 编写了一个简单的程序，它可以在 eclipse nsight 上运行。这是源代码：

#include <iostream>
#include <stdio.h>


__global__ void add( int a,int b, int *c){
*c = a + b;
}

int main(void){

int c;
int *dev_c;

cudaMalloc((void**)&dev_c, sizeof(int));

add <<<1,1>>>(2,7,dev_c);

cudaMemcpy(&c, dev_c, sizeof(int),cudaMemcpyDeviceToHost);

printf("\n2+7= %d\n",c);
cudaFree(dev_c);

return 0;
}

现在我正在尝试将这段代码与 Go 语言和 cgo 一起使用！ 所以我写了这个新代码：

package main

//#include "/usr/local/cuda-7.0/include/cuda.h"
//#include "/usr/local/cuda-7.0/include/cuda_runtime.h"
//#cgo LDFLAGS: -lcuda
//#cgo LDFLAGS: -lcurand
////default location:
//#cgo LDFLAGS: -L/usr/local/cuda-7.0/lib64 -L/usr/local/cuda-7.0/lib
//#cgo CFLAGS: -I/usr/local/cuda-7.0/include/
//
//
//
//
//
//
//
//
//
//
/*

#include <stdio.h>

__global__ void add( int a,int b, int *c){
    *c = a + b;
}

int esegui_somma(void){

    int c;
    int *dev_c;

    cudaMalloc((void**)&dev_c, sizeof(int));
    add <<<1,1>>> (2,7,dev_c);
    cudaMemcpy(&c, dev_c, sizeof(int),cudaMemcpyDeviceToHost);

    cudaFree(dev_c);
    return c;
}
*/
import "C"
import "fmt"

func main(){
    fmt.Printf("il risultato è %d",C.esegui_somma)
}

但它不起作用！ 我读到了这个错误消息：

cgo_cudabyexample_1/main.go:34:8: error: expected expression before '<' token
add <<<1,1>>> (2,7,dev_c);
      ^

我认为我必须为 cgo 而不是 gcc 设置 nvcc cuda 编译器。 我该怎么做？我可以更改CC环境变量吗？ 此致

我终于想出了如何做到这一点。事情最大的问题是nvcc不遵循gcc标准标志和不同clang它不会默默地忽略它们。cgo通过添加一堆用户未明确指定的标志来触发问题。

为了使其全部工作，您需要将设备代码和直接调用它的函数分离到单独的文件中，并直接使用编译/打包它们nvcc到共享库（.so）中。然后，您将使用 cgo 使用系统上的任何默认链接器来链接此共享库。你唯一需要添加的是-lcudart给你的LDFLAGS（链接器标志）链接 CUDA 运行时。