我正在尝试从几个 .cu 源文件和一个非常简单的 C++ main 构建一个简单的应用程序,它从其中一个 .cu 文件调用函数。我正在从编译的 .cu 文件创建一个共享库(.so 文件)。我发现一切都没有问题,但是当我尝试运行应用程序时,我收到一个链接器未定义的符号错误,其中包含我从main()调用的.cu函数的名称。如果我构建一个静态库,我的应用程序运行得很好。这是我设置的生成文件:
.PHONY: clean
NVCCFLAGS = -std=c++11 --compiler-options '-fPIC'
CXXFLAGS = -std=c++11
HLIB = libhello.a
SHLIB = libhello.so
CUDA_OBJECTS = bridge.o add.o
all: driver
%.o :: %.cu
nvcc -o $@ $(NVCCFLAGS) -c -I. $<
%.o :: %.cpp
c++ $(CXXFLAGS) -o $@ -c -I. $<
$(HLIB): $(CUDA_OBJECTS)
ar rcs $@ $^
$(SHLIB): $(CUDA_OBJECTS)
nvcc $(NVCCFLAGS) --shared -o $@ $^
#driver : driver.o $(HLIB)
# c++ -std=c++11 -fPIC -o $@ driver.o -L. -lhello -L/usr/local/cuda-10.1/targets/x86_64-linux/lib -lcudart
driver : driver.o $(SHLIB)
c++ -std=c++11 -fPIC -o $@ driver.o -L. -lhello
clean:
-rm -f driver *.o *.so *.a
以下是生成文件作为素材的各种源文件。 add.cu:
__global__ void add(int n, int* a, int* b, int* c) {
int index = threadIdx.x;
int stride = blockDim.x;
for (int ii = index; ii < n; ii += stride) {
c[ii] = a[ii] + b[ii];
}
}
添加 h:
extern __global__ void add(int n, int* a, int* b, int* c);
bridge.cu:
#include <iostream>
#include "add.h"
void bridge() {
int N = 1 << 16;
int blockSize = 256;
int numBlocks = (N + blockSize - 1)/blockSize;
int* a;
int* b;
int* c;
cudaMallocManaged(&a, N*sizeof(int));
cudaMallocManaged(&b, N*sizeof(int));
cudaMallocManaged(&c, N*sizeof(int));
for (int ii = 0; ii < N; ii++) {
a[ii] = ii;
b[ii] = 2*ii;
}
add<<<numBlocks, blockSize>>>(N, a, b, c);
cudaDeviceSynchronize();
for (int ii = 0; ii < N; ii++) {
std::cout << a[ii] << " + " << b[ii] << " = " << c[ii] << std::endl;
}
cudaFree(a);
cudaFree(b);
cudaFree(c);
}
桥.h:
extern void bridge();
驱动程序.cpp:
#include "bridge.h"
int main() {
bridge();
return 0;
}
我对 cuda 很陌生,所以我希望这就是我做错的地方。我已经尝试了使用 extern "C" 声明,但这似乎只是将"未定义的符号"错误从运行时移动到构建时。
我熟悉各种方法,最终得到一个未定义的符号,我已经提到了我已经执行过的各种实验(静态链接,外部"C"声明),这些实验让我认为这个问题没有通过提议的重复问题来解决。
我的未解析符号是_Z6bridgev
在我看来,链接器应该能够解析该符号。如果我可以在 driver.o 上 nm,我看到:
0000000000000000 T main
U _Z6bridgev
如果我在 libhello.so 上运行 nm,我会看到:
0000000000006e56 T _Z6bridgev
当罗伯特·克罗维拉(Robert Crovella)能够让我的例子在他的机器上工作时,而我却无法让他的例子在我的机器上工作,我开始意识到我的问题与cuda或nvcc无关。事实上,对于共享库,加载器必须在运行时解析符号,而我的共享库不在"众所周知的位置"。我刚才构建了一个简单的测试用例,纯粹使用 c++ 源代码,并重复了我的失败。一旦我libhello.so
复制到/usr/local/lib
,我就能成功地运行driver
。所以,我可以结束我最初的问题,如果这是人民的意愿。