如何從CUDA內核函數返回一個變量？

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); 

對於這兩個行我得到這個錯誤： 錯誤：類型「長」的說法是有類型的「常量無效*」

Answer 1

爲了得到一個結果的參數不符，你必須的memcpy它，即︰

#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; 
} 

我想錯誤來，因爲你傳遞一個長的值，而不是一個長指針指針。

Answer 2

我一直在使用__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; 
}