PTX内核名称损坏

Question

当从ptx文件编译内核时，我无法链接我的Cuda程序。

main.cu:

extern 
__global__ void kernel(int, float*);

int main()
{
    ...
    kernel<<<...>>>(...);
    ...
}

kernel.cu

__global__
void kernel(int n, float* p)
{
    ...
}

如果我像下面这样编译，我没有问题，我得到一个可执行文件：

nvcc -dc main.cu kernel.cu --gpu-architecture=sm_70
nvcc -dlink main.o kernel.o --gpu-architecture=sm_70 -o dlink.o
g++ dlink.o main.o kernel.o -lcudart

如果我像下面这样编译(通过生成ptx)，我会得到错误：

nvcc -ptx kernel.cu --gpu-architecture=sm_70
nvcc -dc main.cu kernel.ptx --gpu-architecture=sm_70
nvcc -dlink main.o kernel.o --gpu-architecture=sm_70 -o dlink.o
g++ dlink.o main.o kernel.o -lcudart

错误：

main.o: In function `main':
tmpxft_0000b5ce_00000000-5_main.cudafe1.cpp:(.text+0x4789): undefined reference to `kernel(int, float*)'
tmpxft_0000b5ce_00000000-5_main.cudafe1.cpp:(.text+0x497e): undefined reference to `kernel(int, float*)'
collect2: error: ld returned 1 exit status

我正在效仿CUDA_Compiler_Driver_NVCC.pdf的一个例子。

我需要做些什么来修复错误？

(这是CUDA 10.2)。

Answer 1

如果您想编写自己的PTX (或修改PTX)，正确的CUDA方法是CUDA驱动程序API和相关的编译流程。

CUDA vectorAddDrv 样本代码拥有您所需的所有管道和工作流。