2012-08-23 62 views
1

我无法在此短cuda代码中获得分段错误的来源。 我正在使用它来测试Thrust库与STL库的排序速度,排序整数为 。我传递的数组大小作为命令行 参数进行排序。在小代码中查找seg错误

下面是代码

inline void check_cuda_error(char *message) 

    { 
     cudaThreadSynchronize(); 
     cudaError_t error = cudaGetLastError(); 
     if(error != cudaSuccess) 
     { 
     printf("CUDA error after %s: %s\n", message, cudaGetErrorString(error)); 
     } 
    } 



      int main(int argc, char *argv[]) 
     { 
      int N = atoi(argv[1]); 
      double* h = new double[N]; 
      for (int i = 0; i < N; ++i) 
      { 
       h[i] = (double)rand()/RAND_MAX; //std::cout << h[i] << " " ; 
      } 

      clock_t start , stop; 

      std::cout << std::endl; 

      // Start timing 
      start = clock(); 
      std::sort(h, h+N); 
      stop = clock(); 
      std::cout << "Host sorting took " << (stop - start) /(double)CLOCKS_PER_SEC << std::endl ; 


      // Start the GPU work. Initialize to random numbers again. 
      for (int i = 0; i < N; ++i) 
      { 
       h[i] = (double)rand()/RAND_MAX; //std::cout << h[i] << " " ; 
      } 
      double* d = 0; 
      const size_t num_bytes = N * sizeof(double); 
      cudaMalloc((void**)&d, num_bytes); 
      check_cuda_error("Memory Allocation"); 

      cudaMemcpy(d ,h , N * sizeof(double), cudaMemcpyHostToDevice); // Transfer data 
      thrust::sort(d, d+ N) ; 
      return 0; 
     } 

我收到以下错误

[BeamerLatex/Farber]$ nvcc -arch=sm_20 sortcompare.cu ; ./a.out 16777216 

Host sorting took 3.77 
[1] 4661 segmentation fault ./a.out 16777216 
[BeamerLatex/Farber]$ 
+0

不相关,但你不删除'h' ... –

+0

你可以添加你的包括?我cba谷歌他们所有,这将有助于我有一个运行版本的代码 – sji

+0

只是一个猜测,但是,你可以请检查cudaMalloc的returnvalue。如果是cudaErrorMemoryAllocation,那么对cudaMemcpy的调用很可能会失败,因为目标仍然是0.这通常会导致段错误。 –

回答

2

看来,你不能在一个原始指针运行推力::排序,你需要将其转换为device_ptr第一,即:

thrust::device_ptr<double> dv = thrust::device_pointer_cast(d); 
thrust::sort(dv, dv+ NN) ; 

这对我很好。

+0

啊谢谢!已经忘记了这一点, – smilingbuddha