cuda - 强制 CUDA 的 thrust::reduce 以非并行方式执行

Question

我有一个使用 thrust::reduce 来并行求和的 CUDA 程序:例如，

thrust::device_ptr<double> tmp(aux);
double my_sum = thrust::reduce(tmp, tmp + G);

其中 double* aux 指向设备上的 G 连续 double 。我需要将整个并行化程序的运行时间与没有并行计算的版本进行比较。有没有办法在设备上仅使用一个线程来运行 thrust::reduce？全局开关将是最方便的选择。

Answer 1

您应该能够通过使用串行执行策略在内核中调用 thrust::reduce 然后使用单个线程启动该内核来实现这一点。像这样的东西:

__global__ void serial_reduce(double *result, double *aux, int G)
{
    *result = thrust::reduce(thrust::seq, aux, aux+G);
}

double *result;
cudaMallocManaged(&result, sizeof(double));
serial_reduce<<<1,1>>>(result, aux, G);
cudaDeviceSynchronize();

[注意在浏览器中编写，完全未经测试，使用风险自负]