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