CUDA Schub-und sort_by_key

Können Sie ändern Vergleichsfunktion zum Sortieren von Schlüsseln anstelle von Werten. @Robert Crovella richtig darauf hingewiesen, dass ein raw-device-pointer zugeordnet werden, kann nicht vom host. Der modifizierte Algorithmus ist unten:

struct cmp : public binary_function<int,int,bool>
{
  cmp(const double *ptr) : rawA(ptr) { }

  __host__ __device__ bool operator()(const int i, const int j) const 
  {return rawA[i] > rawA[j];}

   const double *rawA; //an array in global mem
}; 

void sortkeys(double *A, int n) {
  //move data to the gpu
  thrust::device_vector<double> devA(A, A + n);
  double *rawA = thrust::raw_pointer_cast(devA.data());

  thrust::device_vector<int> B(n);
  //initialize keys
  thrust::sequence(B.begin(), B.end());
  thrust::sort(B.begin(), B.end(), cmp(rawA));
  //B now contains the sorted keys
 }

Und hier ist die alternative mit arrayfire. Obwohl ich nicht sicher bin, welches effizienter ist, da arrayfire Lösung verwendet zwei zusätzliche Felder:

void sortkeys(double *A, int n) {
   af::array devA(n, A, af::afHost);
   af::array vals, indices;
   //sort and populate vals/indices arrays
   af::sort(vals, indices, devA);
   std::cout << devA << "\n" << indices << "\n";
}

Aufbauend auf der Antwort von @asm (ich war nicht in der Lage, um es arbeiten), dieser code schien, für mich zu arbeiten, und nicht nur Sortieren der keys. Ich glaube jedoch, es beschränkt sich auf den Fall, wo die Schlüssel sind in der Reihenfolge 0, 1, 2, 3, 4 ... entsprechend der (Doppel -) Werte. Da dies ein "index-Wert" Sortieren, es kann erweitert werden auf den Fall einer beliebigen Reihenfolge von Tasten, vielleicht, indem Sie ein indiziertes kopieren. Aber ich bin nicht sicher, dass der Prozess der Generierung der index-Reihenfolge und dann die Umgestaltung des original-keys werden nicht schneller, als nur das kopieren der ursprünglichen Wert auf einen neuen Vektor (für den Fall der willkürlichen Schlüssel).

#include <iostream>
#include <thrust/device_vector.h>
#include <thrust/host_vector.h>
#include <thrust/sort.h>

using namespace std;

__device__  double *rawA; //an array in global mem

struct cmp : public binary_function<int, int, bool>
{
  __host__ __device__  bool operator()(const int i, const int j) const
  {return ( rawA[i] < rawA[j]);}
};

void sortkeys(double *A, int n) {
  //move data to the gpu
  thrust::device_vector<double> devA(A, A + n);
// rawA = thrust::raw_pointer_cast(&(devA[0]));
  double *test = raw_pointer_cast(devA.data());
  cudaMemcpyToSymbol(rawA, &test, sizeof(double *));

  thrust::device_vector<int> B(n);
  //initialize keys
  thrust::sequence(B.begin(), B.end());
  thrust::sort(B.begin(), B.end(), cmp());
  //B now contains the sorted keys
  thrust::host_vector<int> hostB = B;
  for (int i=0; i<hostB.size(); i++)
    std::cout << hostB[i] << " ";
  std::cout<<std::endl;
  for (int i=0; i<hostB.size(); i++)
    std::cout << A[hostB[i]] << " ";
  std::cout<<std::endl;
 }


int main(){

  double C[] = {0.7, 0.3, 0.4, 0.2, 0.6, 1.2, -0.5, 0.5, 0.0, 10.0};
  sortkeys(C, 9);
  std::cout << std::endl;
  return 0;
}