Golang CUDA kitaplığını çağırıyor

Question

Go kodumdan bir CUDA işlevini çağırmaya çalışıyorum. Aşağıdaki üç dosyam var.

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()) 
} 

Birlikte CUDA kodu derleme am:

nvcc -m64 -arch=sm_20 -o libtest.so --shared -Xcompiler -fPIC test.cu 

Tüm üç dosya - test.h, test.cu ve test.go aynı dizinde. Gitmeye çalıştığımda aldığım hata "test_add" için tanımlanmamış başvuru "dır.

C/C++ ile çok az deneyimim var ve CUDA'da yeni bir acemiyim.

Sorunumu iki gün boyunca çözmeye çalışıyorum ve herhangi bir girdi için çok minnettar olacağım.

Teşekkürler.

Answer 1

En azından bu durumda, bu the go import of C is expecting the function to be provided with C style linkage görünür.

CUDA (yani nvcc) temel olarak C++ desen izler ve varsayılan C++ stili bağlantı (fonksiyon Cendereleme vb dahil olmak üzere)

Bu C yerine C kullanarak dışarıdan sağlanabilir bir kod bölümü kalmasını sağlamak mümkündür tarafından içerir ++ extern "C" {...code...} kullanarak stil bağlantısı. Bu C++ dil özelliğidir ve CUDA veya nvcc'ye özgü değildir.

extern "C" { int test_add(void) { ... code ... }; }