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と関連するコンパイルフローです。

CUDAvectorAddDrv サンプルコードには、必要なすべての配管とワークフローが含まれています。