2

我一直在尝试编写一个函数，该函数将使用加速库获取向量的直方图。我认识到直方图不是 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 获取直方图？

4

1 回答 1

1

这是 Accelerate（和/或 CUDA 中的潜在变化）中的一个错误，现已修复。很抱歉花了这么长时间才找到它，这从我的雷达上溜走了。

于 2016-08-15T03:23:55.427 回答