编译和链接纯 C 和 CUDA 代码 [警告:函数的隐式声明]

Compiling and Linking pure C and CUDA code [warning: implicit declaration of function]

我正在尝试编译 link .c 和 .cu 文件,但收到警告

 warning: implicit declaration of function

我在 .cu 文件中有一个函数需要从 .c 文件中调用。 .c 文件使用 gcc 编译,.cu 文件使用 nvcc 编译器编译。由于 .cu 文件的头文件包含内置的 cuda 数据类型,我不能将其包含在 .c 文件中。我仍然能够编译和 link 所有文件,但我想摆脱我无法做到的警告。代码的基本结构是:

gpu.cu
    void fooInsideCuda();

cpu.c
    fooInsideCuda(); //calling function in gpu.cu

如有任何帮助或建议,我们将不胜感激。

这个link:https://devtalk.nvidia.com/default/topic/388072/calling-cuda-functions-from-a-c-file/

回答您的问题:,。基本上:

在.c文件中

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <cuda.h>

extern void kernel_wrapper(int *a, int *b);

int main(int argc, char *argv[])
{
    int a = 2;
    int b = 3;

    kernel_wrapper(&a, &b);
    return 0;
}

并在 .cu 文件中;

__global__ void kernel(int *a, int *b)
{
    int tx = threadIdx.x;

    switch( tx )
    {
    case 0:
     *a = *a + 10;
     break;
    case 1:
     *b = *b + 3;
     break;
    default:
     break;
    }

}

void kernel_wrapper(int *a, int *b)
{
    int *d_1, *d_2;

    dim3 threads( 2, 1 );
    dim3 blocks( 1, 1 );

    cudaMalloc( (void **)&d_1, sizeof(int) );
    cudaMalloc( (void **)&d_2, sizeof(int) );

    cudaMemcpy( d_1, a, sizeof(int), cudaMemcpyHostToDevice );
    cudaMemcpy( d_2, b, sizeof(int), cudaMemcpyHostToDevice );

    kernel<<< blocks, threads >>>( a, b );

    cudaMemcpy( a, d_1, sizeof(int), cudaMemcpyDeviceToHost );
    cudaMemcpy( b, d_2, sizeof(int), cudaMemcpyDeviceToHost );

    cudaFree(d_1);
    cudaFree(d_2);
}

然后是类似这样的 .h 文件:

#ifndef __B__
#define __B__

#include "cuda.h"
#include "cuda_runtime.h"

extern "C" void kernel_wrapper(int *a, int *b);
#endif

另请注意,.cu 编译器使用 C++ 约定

所以在 .cu 文件中需要如下内容:

extern "C" void A(void)
{
    .......
}

因此使用了 'C' 约定