从NVIDIA documentation开始,当生成PTX,CUBIN或FATBIN时,主机代码将从文件中丢弃。现在我有我的主机代码(main.cu)和设备代码(shared.cu)。使用nvcc选项*.o
或使用nvcc -c main.cu shared.cu
将每个文件编译为nvcc -dc main.cu shared.cu
并将其与选项nvcc -link main.o shared.o
相关联时,我可以生成可执行文件。但是,当shared.cu
编译为shared.cubin
并进一步编译为*.o
时,链接将失败并显示错误tmpxft_00001253_00000000-4_main.cudafe1.cpp:(.text+0x150): undefined reference to <KERNEL FUNCTION>
在此我想知道shared.cu
只包含设备代码,即使删除了主机代码,为什么链接也会失败。
源代码文件是 main.cu
#include <stdio.h>
#include <cuda_runtime_api.h>
#include <cuda_runtime.h>
#include <cuda.h>
#include "shared.h"
int main()
{
int a[5]={1,2,3,4,5};
int b[5]={1,1,1,1,1};
int c[5];
int i;
int *dev_a;
int *dev_b;
int *dev_c;
cudaMalloc( (void**)&dev_a, 5*sizeof(int) );
cudaMalloc( (void**)&dev_b, 5*sizeof(int) );
cudaMalloc( (void**)&dev_c, 5*sizeof(int) );
cudaMemcpy(dev_a, a , 5 * sizeof(int), cudaMemcpyHostToDevice);
cudaMemcpy(dev_b, b , 5 * sizeof(int), cudaMemcpyHostToDevice);
add<<<1,5>>>(dev_a,dev_b,dev_c);
cudaMemcpy(&c,dev_c,5*sizeof(int),cudaMemcpyDeviceToHost);
for(i = 0; i < 5; i++ )
{
printf("a[%d] + b[%d] = %d\n",i,i,c[i]);
}
cudaFree( dev_a);
cudaFree( dev_b);
cudaFree( dev_c);
return 0;
}
shared.cu
#include<stdio.h>
__global__ void add(int *dev_a, int *dev_b, int *dev_c){
//allocate shared memory
__shared__ int a_shared[5];
__shared__ int b_shared[5];
__shared__ int c_shared[5];
{
//get data in shared memory
a_shared[threadIdx.x]=dev_a[threadIdx.x];
__syncthreads();
b_shared[threadIdx.x]=dev_b[threadIdx.x];
__syncthreads();
//perform the addition in the shared memory space
c_shared[threadIdx.x]= a_shared[threadIdx.x] + b_shared[threadIdx.x];
__syncthreads();
//shift data back to global memory
dev_c[threadIdx.x]=c_shared[threadIdx.x];
__syncthreads();
}
}
shared.h
#ifndef header
#define header
extern __global__ void add(int *dev_a, int *dev_b, int *dev_c);
#endif
答案 0 :(得分:2)
我相信您假设“仅设备代码”文件(例如您的shared.cu
)不包含主机代码。这实际上并不正确。
内核函数生成主机和设备代码构造,这些构造由CUDA预处理器(cudafe)生成并拆分。请参阅the documentation中的CUDA编译轨迹。
请注意,主机和设备代码之间存在初始分离,然后是.cudafe1.stub.c文件的后续创建,然后将其传递到主机端(即与Cubin路径分开,实际上是启用链接)。
如该图所示,.cudafe1.stub.c文件不会成为cubin的一部分,而是进入主机端处理流,最终成为fatbinary文件的一部分。
如果只处理cubin,则丢弃此.cudafe1.stub.c,这对于最终链接创建可执行的胖二进制文件是必要的。
因此,由于存根文件中缺少引用,尝试使用只有cubin创建胖二进制链接将失败,该引用将在您的问题的错误输出中指示。