我正在尝试从我的Go代码中调用CUDA函数。 我有以下三个文件。
test.h:
int test_add(void);
test.cu:
__global__ void add(int *a, int *b, int *c){
*c = *a + *b;
}
int test_add(void) {
int a, b, c; // host copies of a, b, c
int *d_a, *d_b, *d_c; // device copies of a, b, c
int size = sizeof(int);
// Allocate space for device copies of a, b, c
cudaMalloc((void **)&d_a, size);
cudaMalloc((void **)&d_b, size);
cudaMalloc((void **)&d_c, size);
// Setup input values
a = 2;
b = 7;
// Copy inputs to device
cudaMemcpy(d_a, &a, size, cudaMemcpyHostToDevice);
cudaMemcpy(d_b, &b, size, cudaMemcpyHostToDevice);
// Launch add() kernel on GPU
add<<<1,1>>>(d_a, d_b, d_c);
// Copy result back to host
cudaMemcpy(&c, d_c, size, cudaMemcpyDeviceToHost);
// Cleanup
cudaFree(d_a); cudaFree(d_b); cudaFree(d_c);
return 0;
}
test.go:
package main
import "fmt"
//#cgo CFLAGS: -I.
//#cgo LDFLAGS: -L. -ltest
//#cgo LDFLAGS: -lcudart
//#include <test.h>
import "C"
func main() {
fmt.Printf("Invoking cuda library...\n")
fmt.Println("Done ", C.test_add())
}
我正在使用以下工具编译CUDA代码:
nvcc -m64 -arch=sm_20 -o libtest.so --shared -Xcompiler -fPIC test.cu
所有三个文件 - test.h,test.cu和test.go都在同一个目录中。 当我尝试使用go构建时,出现的错误是“未定义对`test_add'的引用”。
我对C/C ++几乎没有经验,CUDA方面完全是新手。
我已经尝试解决这个问题两天了,非常感谢任何帮助。
谢谢。
test_add()
原型用extern "C" { ... }
包装起来。 - Robert CrovellaC
期望以C风格的链接提供函数,那么你就会遇到这个问题。这只是一个猜测。 - Robert Crovella