【发布时间】：2011-11-04 23:28:53
【问题描述】：
我正在尝试在 cuda/thrust 中编写以下问题。我得到一个键列表和与每个键关联的三个值。我设法按字典顺序对它们进行了排序。如果具有相同键的输入具有每个值关系，则现在需要减少输入。在下面的示例中，V1(a)
示例输入：
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
- 输入 a c 已删除
- 输入 a d 已删除
我已经能够使用 for 循环和 if 语句解决上述问题。我目前正在尝试使用基于 gpu 的 cuda/thrust 来解决这个问题。这可以在 gpu 上完成（最好是推力）还是必须用 cuda 编写单个内核？
我没有像 Thrust: Removing duplicates in key-value arrays 中讨论的那样使用 unique 来制定这个问题
已编辑以包含程序“stl/c++”程序以生成上述场景：“Reducing 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;
}
【问题讨论】：
-
无法准确识别您需要什么，但似乎thrust::unique_by_key 可能是解决方案。您能否发布您的串行解决方案的链接？
-
我不认为
thrust::unique_by_key会以简单的方式工作，因为它通过查看相邻元素来完成独特的步骤。解决方案需要查看所有可能的配对。