尝试与从 CUDA 对象构建的共享库链接时未定义的符号



我正在尝试从几个 .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。所以,我可以结束我最初的问题,如果这是人民的意愿。

最新更新