Question

我有一个设备矢量A，B，C如下。

A = [1,1,3,3,3,4,4,5,5]
B = [1,3,5]
C = [2,8,6]

所以我想用C中的相应元素替换每个B中的B。例如：

1替换为2，
3替换为8，
5由6

以获得以下结果

Result = [2,2,8,8,8,4,4,6,6]

我如何在cuda推力中实现这一目标或以任何方式在cuda C ++中实现它。我发现thrust :: replace可以立即替换单个元素。由于我需要替换大量数据，因此一次更换一个数据就成了瓶颈。

Answer 1

这可以通过首先构建地图然后应用查询地图的自定义函子来有效地完成。

示例代码执行以下步骤：

获取.main { z-index: 1; /* this is causing your problems */ }的最大元素。这假定您的数据已经排序。
创建大小为C的地图矢量。将新值复制到旧值的位置。
将largest_element仿函数应用于mapper。该仿函数从地图矢量中读取A。如果此new_value 不 new_value，则0中的值将替换为新值。这假定A永远不会包含C。如果它可以包含0，则必须使用其他条件，例如使用0初始化地图矢量，然后检查-1

new_value != -1

<强>输出

#include <thrust/device_vector.h>
#include <thrust/iterator/permutation_iterator.h>
#include <thrust/copy.h>
#include <thrust/for_each.h>
#include <thrust/scatter.h>
#include <iostream>


#define PRINTER(name) print(#name, (name))
template <template <typename...> class V, typename T, typename ...Args>
void print(const char* name, const V<T,Args...> & v)
{
    std::cout << name << ":\t";
    thrust::copy(v.begin(), v.end(), std::ostream_iterator<T>(std::cout, "\t"));
    std::cout << std::endl;
}


template <typename T>
struct mapper
{
    mapper(thrust::device_ptr<const T> map) : map(map)
    {
    }

    __host__ __device__
    void operator()(T& value) const
    {
       const T& new_value = map[value]; 
       if (new_value)
       {
          value = new_value;
       }
    }

    thrust::device_ptr<const T> map;
};

int main()
{
    using namespace thrust::placeholders;

    int A[] = {1,1,3,3,3,4,4,5,5};
    int B[] = {1,3,5};
    int C[] = {2,8,6};

    int size_data    = sizeof(A)/sizeof(A[0]);
    int size_replace = sizeof(B)/sizeof(B[0]);

    // copy demo data to GPU
    thrust::device_vector<int> d_A (A, A+size_data);
    thrust::device_vector<int> d_B (B, B+size_replace);
    thrust::device_vector<int> d_C (C, C+size_replace);

    PRINTER(d_A);
    PRINTER(d_B);
    PRINTER(d_C);

    int largest_element = d_C.back();

    thrust::device_vector<int> d_map(largest_element);

    thrust::scatter(d_C.begin(), d_C.end(), d_B.begin(), d_map.begin());
    PRINTER(d_map);

    thrust::for_each(d_A.begin(), d_A.end(), mapper<int>(d_map.data()));
    PRINTER(d_A);

    return 0;
}

在库达推力中取代多项目

1 个答案: