cuda - Thrust::min_element 在 Thrust::device_vector 上发生崩溃(CUDA Thrust)

Question

以下 CUDA Thrust 程序崩溃:

#include <thrust/device_vector.h>
#include <thrust/extrema.h>

int main(void)
{
  thrust::device_vector<int> vec;
  for (int i(0); i < 1000; ++i) {
    vec.push_back(i);
  }

  thrust::min_element(vec.begin(), vec.end());
}

我得到的异常(exception)是:

Unhandled exception at 0x7650b9bc in test_thrust.exe: Microsoft C++
exception:thrust::system::system_error at memory location 0x0017f178..

In `checked_cudaMemcpy()` in `trivial_copy.inl`.

如果我添加#include <thrust/sort.h>并替换min_element与 sort ，它不会崩溃。

我在 Windows 7 64 位、compute_20、sm_20 (Fermi)、调试版本上使用 CUDA 4.1。在发布版本中，我没有遇到崩溃，并且 min_element 找到了正确的元素。

我做错了什么，还是 Thrust 存在错误？

Answer 1

我可以使用针对计算能力 2.0 的 Debug模式重现该错误(即 nvcc -G0 -arch=sm_20)。该错误在 Release模式下或针对计算功能 1.x 设备时不会重现，这通常表明存在代码生成问题，而不是库中的错误。无论问题出在哪里，我都鼓励您 submit a bug report所以这个问题得到了应有的关注。同时，我建议在 Release模式下进行编译，这样会经过更严格的测试。