Modifica del nome del kernel PTX

Question

Non riesco a collegare il mio programma Cuda quando un kernel è compilato da ptxfile.

main.cu:

extern 
__global__ void kernel(int, float*);

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

kernel.cu

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

Se compilo come di seguito, non ho problemi e ottengo un eseguibile:

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

Se compilo come di seguito (generando ptx), ottengo errori:

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

Errore:

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

Sto seguendo un esempio da CUDA_Compiler_Driver_NVCC.pdf.

Cosa devo fare per correggere l'errore?

(Questo è CUDA 10.2).

Answer 1

Se si desidera scrivere il proprio PTX (o modificare il PTX), la metodologia CUDA corretta da utilizzare è l'API del driver CUDA e il flusso di compilazione associato.

Il vectorAddDrv codice di esempio CUDA contiene tutte le tubature e il flusso di lavoro di cui hai bisogno.