如何将稠密矢量转换为CUDA中的稀疏矢量？

Question

我在GPU存储器大的致密的载体（未矩阵）：

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

，并希望将其转换成稀疏格式：

values = [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];