1. 06 Sep, 2022 1 commit
    • Chang Liu's avatar
      [Feature] Unify the cuda stream used in core library (#4480) · 1c9d2a03
      Chang Liu authored
      
      
      * Use an internal cuda stream for CopyDataFromTo
      
      * small fix white space
      
      * Fix to compile
      
      * Make stream optional in copydata for compile
      
      * fix lint issue
      
      * Update cub functions to use internal stream
      
      * Lint check
      
      * Update CopyTo/CopyFrom/CopyFromTo to use internal stream
      
      * Address comments
      
      * Fix backward CUDA stream
      
      * Avoid overloading CopyFromTo()
      
      * Minor comment update
      
      * Overload copydatafromto in cuda device api
      Co-authored-by: default avatarxiny <xiny@nvidia.com>
      1c9d2a03
  2. 01 Aug, 2022 1 commit
  3. 26 Jul, 2022 1 commit
  4. 11 Apr, 2022 1 commit
  5. 28 Feb, 2022 1 commit
  6. 15 Dec, 2021 1 commit
    • 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