• lixiaobai's avatar
    [PinSAGESampler] support PinSAGE sampler on GPU (#3567) · dd762a1e
    lixiaobai authored
    
    
    * Feat: support API "randomwalk_topk" in library
    
    * Feat: use the new API "randomwalk_topk" for PinSAGESampler
    
    * Minor
    
    * Minor
    
    * Refactor: modified codes as checker required
    
    * Minor
    
    * Minor
    
    * Minor
    
    * Minor
    
    * Fix: checking errors in RandomWalkTopk
    
    * Refactor: modified the docstring for randomwalk_topk
    
    * change randomwalk_topk to internal
    
    * fix
    
    * rename
    
    * Minor for pinsage.py
    
    * Feat: support randomwalk and SelectPinSageNeighbors on GPU
    
    Port RandomWalk algorithm on GPU,
    and port SelectPinSageNeighbors on GPU.
    
    * Feat: support GPU on python APIs
    
    * Feat: remove perf print information in FrequenchHashmap
    
    * Fix: modified the code format
    
    Modified the code format as task_lint.sh suggested
    
    * Feat: let test script support PinSAGESampler on GPU
    
    Let test script support PinSAGESampler on GPU,
    minor of "restart_prob".
    
    * Minor
    
    * Minor
    
    * Minor
    
    * Refactor: use the atomic operations from the array module
    
    * Minor: change the long lines
    
    * Refactor: modified the get_node_types for gpu
    
    * Feat: update the contributor date
    
    * Perf: remove unnecessary stream sync
    
    * Feat: support other random walk
    
    But the non-uniform choice is still not supported.
    
    * Fix: add CUDA switch for random walk
    Co-authored-by: default avatarQuan Gan <coin2028@hotmail.com>
    dd762a1e
randomwalks.cc 6.98 KB