使用C++程序中的C函数 - extern关键字问题

Question

我遇到了C++程序的问题.我想使用在C++文件中的C文件中定义的函数.这是我的C++文件代码:

#include <string>
#include <iostream>
#include <stdio.h>
extern void squre_array();
using namespace std;

int main() {
    squre_array();
}

现在这里是我定义的C文件的代码squre_array():

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

__global__ void square_array(float *a, int N)
{
  int idx = blockIdx.x * blockDim.x + threadIdx.x;
  if (idx<N) 
    a[idx] = a[idx] * a[idx];
}
void squre_array()
{
  float *a_h, *a_d; 
  const int N = 10;  
  size_t size = N * sizeof(float);
  a_h = (float *)malloc(size);        
  cudaMalloc((void **) &a_d, size);   
  for (int i=0; i<N; i++) a_h[i] = (float)i;
  cudaMemcpy(a_d, a_h, size, cudaMemcpyHostToDevice);
  int block_size = 4;
  int n_blocks = N/block_size + (N%block_size == 0 ? 0:1);
  square_array <<< n_blocks, block_size >>> (a_d, N);

  cudaMemcpy(a_h, a_d, sizeof(float)*N, cudaMemcpyDeviceToHost);
  // Print results
  for (int i=0; i<N; i++) printf("%d %f\n", i, a_h[i]);

  free(a_h); 
  cudaFree(a_d);
}

现在任何人都可以告诉我如何将此功能链接到我的C++程序？每次编译程序时,都会收到错误消息:

cpp:: undefined reference to `squre_array()'

谁能告诉我我做错了什么？如何将该squre_array()功能链接到我的C++程序？

Answer 1

在C++代码中,您需要将C函数声明为extern"C"

extern "C" void squre_array();