推力排序示例中的崩溃

Question

我正在尝试官网示例https://developer.nvidia.com/thrust的第一个示例并将向量大小更改为32

#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/generate.h>
#include <thrust/sort.h>
#include <thrust/copy.h>
#include <algorithm>
#include <cstdlib>
#include <time.h>

using namespace std;

int main(void){
  // generate random numbers serially
  thrust::host_vector<int> h_vec(32 << 23);
  std::generate(h_vec.begin(), h_vec.end(), rand);
  std::cout << "1." << time(NULL) << endl;

  // transfer data to the device
  thrust::device_vector<int> d_vec = h_vec;
  cout << "2." << time(NULL) << endl;
  // sort data on the device (846M keys per second on GeForce GTX 480)
  thrust::sort(d_vec.begin(), d_vec.end());
  // transfer data back to host
  thrust::copy(d_vec.begin(), d_vec.end(), h_vec.begin());
  std::cout << "3." << time(NULL) << endl;

  return 0;
}

但是程序在运行到thrust::sort 行时崩溃了。我尝试交替使用 std::vector 和 std:sort 并且效果很好。

这是推力的错误吗？我正在使用 Thrust 1.7 + Cuda 6.5 + Visual Studio 2013 Update 2。

我使用的是 GeForce GT 740M，总内存为 2048M。

我用processexplorer监控进程，看到它分配了1.0G内存。但是我有 2G GPU 内存，16G 主 CPU 内存。

错误消息是“一个问题导致程序停止正常工作。Windows 将关闭程序并通知您是否有可用的解决方案。[调试] [关闭程序]”。单击[Debug]后，我可以看到调用堆栈。问题出在这一行：

thrust::device_vector<int> d_vec = h_vec;

cuda 的最后一个来源是这样的：

testcuda.exe!thrust::system::cuda::detail::malloc<thrust::system::cuda::detail::tag>(thrust::system::cuda::detail::execution_policy<thrust::system::cuda::detail::tag> & __formal, unsigned __int64 n) Line 48  C++

这似乎是一个内存分配问题。但是我有 2G GPU 内存，16G 主 CPU 内存。为什么？？

致罗伯特：

原始示例运行良好，即使对于 32

我的测试代码在这里：https://github.com/henrywoo/wufuheng/blob/master/testcuda.cu

在我的测试中，没有异常，只是运行时出错。

Answer 1

sizeof(int) * 32 即您正在分配大约 1 GB 的 GPU RAM。很可能，您的卡无法处理那么多元素。这可能是因为：

• GPU RAM 通常不足
• 没有足够的连续空闲 GPU RAM（这是必需的，因为向量必须适合连续的内存块）