cuda - 计算 2D 数组 CUDA 的平均值

Question

我需要使用 CUDA 计算 2D 数组的平均值，但我不知道如何继续。我首先进行列缩减，然后对结果数组求和，最后一步我将计算平均值。

为此，我需要立即在设备上完成所有工作吗？或者我只是一步一步地进行，每一步都需要在 CPU 和 GPU 之间来回切换。

Answer 1

如果它是二维数组中所有元素的简单算术平均值，您可以使用推力:

int* data;
int num;
get_data_from_library( &data, &num );

thrust::device_vector< int > iVec(data, data+num);

// transfer to device and compute sum
int sum = thrust::reduce(iVec.begin(), iVec.end(), 0, thrust::plus<int>());
double mean = sum/(double)num;

如果您想编写自己的内核 - 请记住，2D 数组本质上是一个分为行大小块的 1D 数组，并通过 SDK“并行缩减”示例:Whitepaper