如何使用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