2011-08-27 54 views
1

我想在cuda/thrust中编码以下问题。我给出了一个关键字列表和三个与每个键关联的值。我已经设法按字典顺序对它们进行排序。如果具有相同键的输入具有每个数值关系,那么现在需要减少输入。在下面的例子中,V1(a)和V1(c)和V2(a)是V2(c)和V3(a),这意味着Input a <输入c,因此输入c从输出中删除。删除排序列表中的条目:在gpu中有效

示例输入:

 Key  V1  V2  V3 
a.  1  2  5  3 
b.  1  2  6  2 
c.  1  2  7  4   
d.  1  3  6  5   
e.  2  8  8  8 
f.  3  1  2  4 

示例输出:

  Key V1 V2 V3 
a.  1 2 5 3 
b.  1 2 6 2 
e.  2 8 8 8 
f.  3 1 2 4 
  • 输入一个<输入的C ==>℃下除去
  • 输入一个<输入d ==> d除去

我已经能够使用for-loops和if语句解决上述问题。我目前正在尝试使用基于GPU的cuda/thrust来解决这个问题。这是否可以在GPU上完成(最好是推力)或者一个单独的内核必须用cuda编写?

我没有制定使用独特的这个问题,因为在Thrust: Removing duplicates in key-value arrays

编辑讨论包括程序“STL/C++”的程序来生成上述方案:部分“减少MYMAP”使用for循环我的实现和if语句。

#include <iostream> 
#include <tr1/array> 
#include <vector> 
#include <algorithm> 

struct mapItem { 
    mapItem(int k, int v1, int v2, int v3){ 
     key=k; 
     std::tr1::array<int,3> v = {v1, v2, v3}; 
     values=v; 
    }; 
    int key; 
    std::tr1::array<int,3> values; 
}; 

struct sortLexiObj{ 
    bool operator()(const mapItem& lhs, const mapItem& rhs){ 
     return lhs.values < rhs.values; 
    } 
}; 
struct sortKey{ 
    bool operator()(const mapItem& lhs, const mapItem& rhs){ 
     return lhs.key < rhs.key; 
    } 
}; 

int main(int argc, char** argv){ 

    std::vector<mapItem> myMap; 

    // Set up initial matrix: 
    myMap.push_back(mapItem(3, 1, 2, 4)); 
    myMap.push_back(mapItem(1, 2, 6, 2)); 
    myMap.push_back(mapItem(1, 2, 5, 3)); 
    myMap.push_back(mapItem(1, 3, 6, 5)); 
    myMap.push_back(mapItem(2, 8, 8, 8)); 
    myMap.push_back(mapItem(1, 2, 7, 4)); 

    std::sort(myMap.begin(), myMap.end(), sortLexiObj()); 
    std::stable_sort(myMap.begin(), myMap.end(), sortKey()); 
    std::cout << "\r\nOriginal sorted Map" << std::endl; 
    for(std::vector<mapItem>::iterator mt=myMap.begin(); mt!=myMap.end(); ++mt){ 
     std::cout << mt->key << "\t"; 
     for(std::tr1::array<int,3>::iterator it=(mt->values).begin(); it!=(mt->values).end(); ++it){ 
      std::cout << *it << " "; 
     } 
     std::cout << std::endl; 
    } 
    ///////////////////////// 

    // Reducing myMap 
    for(std::vector<mapItem>::iterator it=myMap.begin(); it!=myMap.end(); ++it){ 
     std::vector<mapItem>::iterator jt=it; ++jt; 
     for (; jt != myMap.end();) { 
      if ( (it->key == jt->key)){ 
       if (it->values.at(0) <= jt->values.at(0) && 
        it->values.at(1) <= jt->values.at(1) && 
        it->values.at(2) <= jt->values.at(2)) { 
        jt = myMap.erase(jt); 
       } 
       else ++jt; 
      } 
      else break; 
     } 
    } 

    std::cout << "\r\nReduced Map" << std::endl; 
    for(std::vector<mapItem>::iterator mt=myMap.begin(); mt!=myMap.end(); ++mt){ 
     std::cout << mt->key << "\t"; 
     for(std::tr1::array<int,3>::iterator it=(mt->values).begin(); it!=(mt->values).end(); ++it){ 
      std::cout << *it << " "; 
     } 
     std::cout << std::endl; 
    } 

    return 0; 
} 
+0

难以确切地确定您需要什么,但它似乎像推力:: unique_by_key可能是解决方案。你可以发布一个链接到你的串行解决方案吗? –

+0

在原始文章中添加代码。 – waqy

+0

我不'想'推力:: unique_by_key'将以一种简单的方式工作,因为它通过查看相邻元素来执行独特的步骤。解决方案需要查看所有可能的配对。 –

回答

1

我认为你可以使用thrust::unique有谓语,因为它在Thrust: Removing duplicates in key-value arrays显示真实。 实际上,我们可以这样做是因为的unique以下特征的:具有相同值的

对于每个组在范围[第一连续元素,最后一个),unique移除所有,但第一元件群组。

所以,你应该定义一个谓词来测试 -equality将返回true对具有相同的密钥和所有值都在第一个元组小元组:

typedef thrust::tuple<int, int, int, int> tuple_t; 
    // a functor which defines your *uniqueness* condition 
    struct tupleEqual 
    { 
     __host__ __device__ 
     bool operator()(tuple_t x, tuple_t y) 
     { 
      return ((x.get<0>() == y.get<0>()) // same key 
       && (x.get<1>() <= y.get<1>()) // all values are smaller 
       && (x.get<2>() <= y.get<2>()) 
       && (x.get<3>() <= y.get<3>())); 
     } 
    }; 

而且您必须将其应用于排序的集合。这样,只有第一个元组(最小)不会被删除。 具有相同键和V1,V2或V3中较大值的元组将产生false,因此它不会被删除。

typedef thrust::device_vector<int>    IntVector; 
    typedef IntVector::iterator       IntIterator; 
    typedef thrust::tuple< IntIterator, IntIterator, IntIterator, IntIterator > IntIteratorTuple; 
    typedef thrust::zip_iterator<IntIteratorTuple> ZipIterator; 

    IntVector keyVector; 
    IntVector valVector1, valVector2, valVector3; 

    tupleEqual predicate; 
    ZipIterator newEnd = thrust::unique(
     thrust::make_zip_iterator(
      thrust::make_tuple(
       keyVector.begin(), 
       valVector1.begin(), 
       valVector2.begin(), 
       valVector3.begin())), 
     thrust::make_zip_iterator(
      thrust::make_tuple(
       keyVector.end(), 
       valVector1.end(), 
       valVector2.end(), 
       valVector3.end())), 
     predicate); 

    IntIteratorTuple endTuple = newEnd.get_iterator_tuple(); 

    keyVector.erase(thrust::get<0>(endTuple), keyVector.end()); 
    valVector1.erase(thrust::get<1>(endTuple), valVector1.end()); 
    valVector2.erase(thrust::get<2>(endTuple), valVector2.end()); 
    valVector3.erase(thrust::get<3>(endTuple), valVector3.end());