cuda - 一个小小的 CUDA 难题

Question

我有数组 A[0...N]的 double和数组 B[0...N]的 int .每B[i]变化在 [0...P] .我只需要计算数组 C[0...P] :

C[j] = SUM( A[i] : B[i] = j)

我不能用 N线程与 atomicAdd()功能，因为它不支持 double ， 据我所知。使用 P 的简单实现线程是高度发散的。有没有更好的办法？

Answer 1

如果我理解正确，您正在尝试对 double 数组进行求和减少 A通过 B 中保存的整数键. Thrust模板库包含 reduce_by_key为此操作。 sum rows示例展示了如何将 reduce_by_key 用于类似的应用程序，尽管它使用计数迭代器来生成 key ，而不是使用用户提供的 key 向量。根据您的需要修改它应该是微不足道的。