CUDA에서 고밀도 벡터를 희소 벡터로 변환하는 방법은 무엇입니까?

Question

나는 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];