1

この短い cuda コードでは、セグメンテーション エラーの原因を特定できません。これを使用して、整数のソートについて、Thrust ライブラリと STL ライブラリのソート速度をテストしています。コマンド ライン引数として、並べ替える double の配列のサイズを渡します。

ここにコードがあります

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]$
4

1 に答える 1

2

スラスト::ソートを生のポインターで実行できないようですdevice_ptr。最初にキャストする必要があります。

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

これは私にとってはうまくいきます。

于 2012-08-23T14:10:54.230 に答える