CUDAで密ベクトルを疎ベクトルに変換するには？

Question

Iは、GPUメモリの大密ベクトル（行列ではない）を有する：

[1,3,0,0,4,0,0]

および疎に変換しますフォーマット：

値= [1,3,4];インデックス= [0,1,4]

私はcuSPARSEでcusparse<t>dense2csc()を呼び出すことができます知っているが、それは、マトリックスのために設計されています、そして、ベクターのための効率的ではないかもしれません。これを行う他の方法はありますか？あるいはCUDAカーネルかもしれません。おかげ

Answer 1

使用thrust::copy_if

int * d_index = [1,3,0,0,4,0,0]; 
int * d_index_compact; 

struct non_negative 
{ 
    __host__ __device__ 
    bool operator()(const int x) 
    { 
     return x >= 0; 
    } 
}; 


thrust::copy_if(thrust::cuda::par, d_index, d_index + this->vocab_size , d_index_compact, non_negative()); // d_index_compact = [1,3,4];