cuda - CUDA推力を使用して配列の欠落値を最後の非欠落値で埋める方法は？

Question

欠落している値のフラグは0なので、必要なものから[0, A, B, 0, 0, C, 0]（[0, A, B, B, B, C, C]以前の欠落していない値が存在しない場合は、0のままにしておきます）。

私はCUDAThrustライブラリを使用していますが、各要素をループせずにこれをすばやく実行する方法があるかどうか疑問に思っていました。

どうもありがとう。

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