cuda - 如何使用 CUDA Thrust 用最后一个非缺失值填充数组中的缺失值？

Question

我的缺失值标志是 0，所以[0, A, B, 0, 0, C, 0]我想要[0, A, B, B, B, C, C]（如果不存在以前的非缺失值，则保留为 0）。

我正在使用 CUDA Thrust 库，并且想知道是否有一种快速的方法可以做到这一点，而无需遍历每个元素。

非常感谢。

Answer 1

似乎运作良好。

#include <thrust/device_vector.h>
#include <thrust/scan.h>
#include <iterator>

template<class T>
struct FillMissing
{
    __host__ __device__ T operator()(const T& res, const T& dat)
    {
        return dat == T(0) ? res : dat;
    }
};

int main()
{
    thrust::device_vector<double> vec(7);
    vec[1] = 2;
    vec[2] = 1;
    vec[5] = 3;

    thrust::inclusive_scan(
            vec.begin(), vec.end(),
            vec.begin(),
            FillMissing<double>());

    thrust::copy(
            vec.begin(), vec.end(),
            std::ostream_iterator<double>(std::cout, " "));
    std::cout << std::endl;
}

输出：

0 2 1 1 1 3 3