使用置换迭代器的推力擦除不起作用

thrust erase with permutation iterator not working

我有一个device_vector A。我有一个地图M。我想用地图M擦除A的元素。 我尝试了以下方式,但它给出了编译错误 "no instance of overloaded function..."

#include <thrust/sequence.h>
#include <thrust/execution_policy.h>
#include <thrust/iterator/permutation_iterator.h>
#include <thrust/fill.h>

void erase_value_using_map( thrust::device_vector<int>& A, thrust::device_vector<int> Map)
{

A.erase(thrust::make_permutation_iterator(A.begin(), Map.begin()),
        thrust::make_permutation_iterator(A.begin(), Map.end()));

}

int main(int argc, char * argv[])
{
thrust::device_vector<int> A(20);
thrust::sequence(thrust::device, A.begin(), A.end(),0);  // x components of the 'A' vectors

thrust::device_vector<int> Map(10);
Map[0]=2;Map[1]=4;Map[2]=8;Map[3]=10;Map[4]=11;Map[5]=13;Map[6]=15;Map[7]=17;Map[8]=19;Map[9]=6;

erase_value_using_map(A, Map);

return 0;
}

错误信息:

error: no instance of overloaded function "thrust::device_vector<T, Alloc>::erase [with T=int, Alloc=thrust::device_malloc_allocator<int>]" matches the argument list
            argument types are: (thrust::permutation_iterator<thrust::detail::normal_iterator<thrust::device_ptr<int>>, thrust::detail::normal_iterator<thrust::device_ptr<int>>>, thrust::permutation_iterator<thrust::detail::normal_iterator<thrust::device_ptr<int>>, thrust::detail::normal_iterator<thrust::device_ptr<int>>>)
            object type is: thrust::device_vector<int, thrust::device_malloc_allocator<int>>

我找到了使用收集(按照 talonmies 的建议)和调整大小的解决方案。

#include <thrust/gather.h>
#include <thrust/device_vector.h>
#include <thrust/execution_policy.h>

int main()
{

thrust::device_vector<int> d_values(10);
d_values[0]=0;d_values[1]=10;d_values[2]=20;d_values[3]=30;d_values[4]=40;
d_values[5]=50;d_values[6]=60;d_values[7]=70;d_values[8]=80;d_values[9]=90;

thrust::device_vector<int> d_map(7);
d_map[0]=0;d_map[1]=2;d_map[2]=4;d_map[3]=6;d_map[4]=8;d_map[5]=1;d_map[6]=3;

thrust::device_vector<int> d_output(10);
thrust::gather(thrust::device,
               d_map.begin(), d_map.end(),
               d_values.begin(),
               d_output.begin());

d_output.resize(d_map.size());

return 0;
}