加速代码通过解释器但在 CUDA 下失败

Question

我一直在尝试编写一个函数，该函数将使用加速库获取向量的直方图。我认识到直方图不是 GPU 处理的理想案例，但我正在从一个小种子生成一个相当大的数据集，如果它可以在将其传输回主内存之前减少到几千字节数组，那就太好了。

我想出的代码如下。它需要一些输出箱，然后创建一个新数组，其中 a[x] 的值是 x 在 xs 中出现的次数

hist :: A.Exp Int -> A.Acc (A.Vector Int) -> A.Acc (A.Vector Int)
hist bins xs = A.permute
               (const (+1))
               (A.fill (A.index1 bins) 0)
               (A.index1 . (xs A.!))
               xs

代码似乎在 Accelerate 解释器下正常运行。但是，如果我尝试通过加速 cuda 调用它，我会收到以下错误消息。

./Data/Array/Accelerate/CUDA/State.hs:85:9: (unhandled): CUDA Exception: unspecified launch failure

我的问题有两个。首先，我在做什么导致 CUDA 失败？其次，有没有更好的方法通过 Accelerate 获取直方图？

Answer 1

这是 Accelerate（和/或 CUDA 中的潜在更改）中的一个错误，它有 now been fixed。抱歉花了这么长时间才知道，这从我的视线中溜走了。