【CUDA 】-Thrust sort&sortbykey

#include 
#include 
#include 
#include 
#include
// includes CUDA
#include 
#include
// includes, project
#include 
#include  // helper functions for SDK examples
//include thrust
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define N (3)  //每个对象的条件属性个数
#define M (12) //保存对象个数
__device__ __managed__ int d_attrSelect[N];//定义全局变量，保存有效的属性选择。
struct Element
{
    int attrval[N];//保存该对象的所有的条件属性值
    bool __host__ __device__ operator<(const Element& e)const
    {
        for (int i = 0; i < N; i++)
        {
            if (d_attrSelect[i] == 0)//为0表示该属性无效
            {
                continue;
            }
            if (attrval[i]e.attrval[i])
            {
                return false;
            }
            if (attrval[i] == e.attrval[i])
            {
                continue;
            }
        }
        return false;
    }
    bool __host__ __device__ operator == (const Element & e)const
    {
        bool res = true;
        for (int i = 0; i < N; i++)
        {
            if (d_attrSelect[i] == 0)//如果为0则无效
                continue;
            if (attrval[i] != e.attrval[i])
            {
                res = false;
                break;
            }
        }
        return res;
    }
};
void Split(const string& src, const string& separator, vector& dest)
{
    string str = src;
    string substring;
    string::size_type start = 0, index;

    do
    {
        index = str.find_first_of(separator, start);
        if (index != string::npos)
        {
            substring = str.substr(start, index - start);
            dest.push_back(substring);
            start = str.find_first_not_of(separator, index);
            if (start == string::npos) return;
        }
    } while (index != string::npos);

    //the last token
    substring = str.substr(start);
    dest.push_back(substring);
}
void InData(const char* fname, thrust::host_vector&all_elments)
{
#ifdef TIME_TEST
    double timeuse;
    StartTimer();
#endif
    ifstream fin(fname);
    if (!fin)
    {
        // NO, abort program
        cerr << "can't open input file \"" << fname << "\"" << endl;
        getchar();
        exit(EXIT_FAILURE);
    }
    string line;
    string separtor = " ";
    while (getline(fin, line))
    {
        //cout << line << endl;
        Element temp_obj;
        vector temp_string;
        Split(line, separtor, temp_string);

        for (int i = 0; i < temp_string.size(); i++)//去掉最后的决策属性
        {
            temp_obj.attrval[i] = stoi(temp_string[i]);
            //cout << temp_obj.attrval[i] << "  ";
        }
        //cout << endl;
        all_elments.push_back(temp_obj);
    }
    fin.close();
#ifdef TIME_TEST
    timeuse = GetTimer();
    std::cerr << "[CPU] InData 读取数据的时间（in second）\t" << timeuse << std::endl;
    StartTimer();
#endif
}
int main()
{
    for (int i = 0; i < N; i++)
    {
        d_attrSelect[i] = 1;
    }
    d_attrSelect[2] = 0;
    for (int i = 0; i  ElementList;
    char s[1024] = "C://Users//hym//Desktop//test.txt";
    InData(s, ElementList);
    cout << " CPU: before sort" << endl;
    for (int i = 0; i < M; i++)
    {
        for (int j = 0; j < N; j++)
        {
            cout << ElementList[i].attrval[j] << "  ";
        }
        cout << endl;
        //printf("CPU Sort before:%d %d %d\n", ElementList[i].attrval[0], ElementList[i].attrval[1], ElementList[i].attrval[2]);
    }
    std::sort(ElementList.begin(), ElementList.end());// cpu 串行排序
    thrust::device_vector dev_element = ElementList;
        thrust::sort(dev_element.begin(), dev_element.end());
        ElementList = dev_element;
    cout << " CPU: After sort" << endl;
    for (int i = 0; i < M; i++)
    {
        for (int j = 0; j < N; j++)
        {
            cout << ElementList[i].attrval[j] << "  ";
        }
        cout << endl;
    }
    thrust::host_vector h_value(M);
    for (int i = 0; i < M; i++)
    {
        h_value[i] = i + 1;
    }
    thrust::device_vector d_value = h_value;
    thrust::sort_by_key(dev_element.begin(), dev_element.end(), d_value.begin());
    ElementList = dev_element;
    h_value = d_value;
    for (int i = 0; i < M; i++)
    {
        cout << "id=" << h_value[i] << " ";
        for (int j = 0; j < N; j++)
        {
            cout << ElementList[i].attrval[j] << "  ";
        }
        cout << endl;
    }
    getchar();
    return 0;
}

my test:

0 0 0
0 0 1
1 0 1
1 1 1
0 0 0 
1 1 2
0 0 0 
1 1 1
1 1 2
0 1 1 
1 0 2
1 1 1

最新的cuda中kernel函数中也可以使用thrust了。
Thrust inside user written kernels

使用仿函数进行排序

////////////////////////////////////////////////////////////////////////////
//
// Copyright 1993-2015 NVIDIA Corporation.  All rights reserved.
//
// Please refer to the NVIDIA end user license agreement (EULA) associated
// with this source code for terms and conditions that govern your use of
// this software. Any use, reproduction, disclosure, or distribution of
// this software and related documentation outside the terms of the EULA
// is strictly prohibited.
//
////////////////////////////////////////////////////////////////////////////

/* Template project which demonstrates the basics on how to setup a project
* example application.
* Host code.
*/
///*
// includes, system
#include 
#include 
#include 
#include 
#include
#include