无法链接从CUBIN中间表示生成的CUDA对象文件

时间:2016-06-21 02:11:16

标签: cuda linker nvidia nvcc ptx

从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

1 个答案:

答案 0 :(得分:2)

我相信您假设“仅设备代码”文件(例如您的shared.cu)不包含主机代码。这实际上并不正确。

内核函数生成主机和设备代码构造,这些构造由CUDA预处理器(cudafe)生成并拆分。请参阅the documentation中的CUDA编译轨迹。

请注意,主机和设备代码之间存在初始分离,然后是.cudafe1.stub.c文件的后续创建,然后将其传递到主机端(即与Cubin路径分开,实际上是启用链接)。

如该图所示,.cudafe1.stub.c文件不会成为cubin的一部分,而是进入主机端处理流,最终成为fatbinary文件的一部分。

如果只处理cubin,则丢弃此.cudafe1.stub.c,这对于最终链接创建可执行的胖二进制文件是必要的。

因此,由于存根文件中缺少引用,尝试使用只有cubin创建胖二进制链接将失败,该引用将在您的问题的错误输出中指示。