"vscode:/vscode.git/clone" did not exist on "2e209c30cf6f2ba42001d0629dc6b7ce354b9a9d"
  • 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
frequency_hashmap.cuh 2.17 KB