使用内核内部的推力减少 CUDA

Question

我想进行并行缩减，但在我的内核内部使用共享内存中的数据。推力库可以做到这一点吗？ 像

int sum = thrust::reduce(myIntArray, myIntArray+numberOfItems, (int) 0, thrust::max_element<int>());

但这在内核中不起作用。可能吗？谢谢。

Answer 1

不，thrust::reduce() 是一个主机函数，如果数据在 GPU 上，则会导致执行 CUDA 内核。

您必须深入研究推力源并找到它用于还原的__device__ 函数。这些可以从您的内核中调用。如果减少的逻辑包含在其他__global__ 内核中，则必须手动将其拼凑在一起才能使用它。