Question

我有一个CUDA搜索功能，可以计算一个变量。我怎么能把它还回来。

__global__ 
void G_SearchByNameID(node* Node, long nodeCount, long start,char* dest, long answer){
    answer = 2;
}

cudaMemcpy(h_answer, d_answer, sizeof(long), cudaMemcpyDeviceToHost);
cudaFree(d_answer);

这两行都出现了这个错误： error：类型为“long”的参数与“const void *”类型的参数不兼容

Answer 1

我一直在使用__device__变量来实现此目的，这样您就不必费心使用cudaMalloc和cudaFree而且您不必将指针传递给一个内核参数，它可以在你的内核中保存一个寄存器来启动。

__device__ long d_answer;

__global__ void G_SearchByNameID() {
  d_answer = 2;
}

int main() {
  SearchByNameID<<<1,1>>>();
  typeof(d_answer) answer;
  cudaMemcpyFromSymbol(&answer, "d_answer", sizeof(answer), 0, cudaMemcpyDeviceToHost);
  printf("answer: %d\n", answer);
  return 0;
}

Answer 2

要获得单个结果，您必须记住它，即：

#include <assert.h>

__global__ void g_singleAnswer(long* answer){ *answer = 2; }

int main(){

  long h_answer;
  long* d_answer;
  cudaMalloc(&d_answer, sizeof(long));
  g_singleAnswer<<<1,1>>>(d_answer);
  cudaMemcpy(&h_answer, d_answer, sizeof(long), cudaMemcpyDeviceToHost); 
  cudaFree(d_answer);
  assert(h_answer == 2);
  return 0;
}

我猜错误来了，因为你传递了一个long值，而不是指向long值的指针。

如何从CUDA内核函数返回单个变量？

2 个答案: