• Xin Yao's avatar
    [Feature] Add CUDA Weighted Neighborhood Sampling (#4064) · 86c81b4e
    Xin Yao authored
    
    
    * add weighted sampling without replacement (A-Chao)
    
    * improve Algorithm A-Chao with block-wise prefix sum
    
    * correctly fill out_idxs
    
    * implement weighted sampling with replacement
    
    * small fix
    
    * merge host-side code of weighted/uniform sampling
    
    * enable unit tests for cuda weighted sampling
    
    * move thrust/cub wrapper to the cmake file
    
    * update docs accordingly
    
    * fix linting
    
    * fix linting
    
    * fix unit test
    
    * Bump external CUB/Thrust versions
    
    * Fix code style and update description of algorithm design
    
    * [Feature] GPU support weighted graph neighbor sampling
    commit by pengqirong(OPPO)
    
    * merge pengqirong's implementation
    
    * revert the change to cub and thrust
    
    * fix linting
    
    * use DeviceSegmentedSort for better performance
    
    * add more comments
    
    * add necessary notes
    
    * add necessary notes
    
    * resolve some comments
    
    * define THRUST_CUB_WRAPPED_NAMESPACE
    
    * fix doc
    Co-authored-by: default avatar彭齐荣 <657017034@qq.com>
    86c81b4e
rowwise_sampling.cu 12.7 KB