问题描述
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
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)。
解决方法
如果您要编写自己的PTX(或修改PTX),则要使用的正确CUDA方法是CUDA驱动程序API和相关的编译流程。
CUDA vectorAddDrv
sample code具有您需要的所有管道和工作流程。