1. 07 Dec, 2022 2 commits
  2. 05 Dec, 2022 1 commit
  3. 01 Dec, 2022 1 commit
  4. 20 Nov, 2022 1 commit
  5. 18 Nov, 2022 1 commit
  6. 17 Nov, 2022 1 commit
    • hummingg's avatar
      less than 2^31−1 nodes then 32-bit (#4912) · 3103f8cb
      hummingg authored
      English Version: However, if a graph contains less than 2^31−1 nodes or edges, one should use 32-bit integers as it leads to better speed and requires less memory.
      3103f8cb
  7. 16 Nov, 2022 1 commit
  8. 15 Nov, 2022 1 commit
  9. 10 Nov, 2022 1 commit
  10. 06 Nov, 2022 2 commits
    • Xin Yao's avatar
      [Feature] Add bfloat16 (bf16) support (#4648) · 96297fb8
      Xin Yao authored
      * add bf16 specializations
      
      * remove SWITCH_BITS
      
      * enable amp for bf16
      
      * remove SWITCH_BITS for cpu kernels
      
      * enbale bf16 based on CUDART
      
      * fix compiling for sm<80
      
      * fix cpu build
      
      * enable unit tests
      
      * update doc
      
      * disable test for CUDA < 11.0
      
      * address comments
      
      * address comments
      96297fb8
    • rudongyu's avatar
      [Transform] Add shortest distance (#4799) · 1d229194
      rudongyu authored
      * shortest dist
      
      * update
      
      * address issues
      
      * update
      
      * fix indent
      
      * lint
      
      * fix indent
      1d229194
  11. 02 Nov, 2022 1 commit
    • ZhenyuLU_Heliodore's avatar
      [NN] Add DegreeEncoder for graph transformer (#4742) · 9d9280cb
      ZhenyuLU_Heliodore authored
      
      
      * Add files via upload
      
      This file will contain several utility modules for Graph Transformer. DegreeEncoder has been implemented in this file now.
      
      * Update graph_transformer.py
      
      * Update nn-pytorch.rst
      
      Add graph_transformer.DegreeEncoder in Utility Modules
      
      * Update test_nn.py
      
      * Update __init__.py
      
      * Update graph_transformer.py
      
      * Update test_nn.py
      
      * Update graph_transformer.py
      
      fix unused import
      
      * Update graph_transformer.py
      
      add module doc-string
      
      * Update graph_transformer.py
      
      * Update graph_transformer.py
      Co-authored-by: default avatarrudongyu <ru_dongyu@outlook.com>
      9d9280cb
  12. 01 Nov, 2022 1 commit
  13. 31 Oct, 2022 1 commit
  14. 27 Oct, 2022 2 commits
  15. 12 Oct, 2022 1 commit
  16. 10 Oct, 2022 1 commit
  17. 08 Oct, 2022 1 commit
  18. 30 Sep, 2022 1 commit
  19. 29 Sep, 2022 1 commit
  20. 28 Sep, 2022 1 commit
  21. 27 Sep, 2022 1 commit
  22. 26 Sep, 2022 1 commit
    • czkkkkkk's avatar
      [Sparse] Add reduction operators (#4600) · 544a2cd1
      czkkkkkk authored
      * [Sparse] Add reduction operators
      
      * Fix linting problem
      
      * Fix linting problem
      
      * Fix linting problem
      
      * Fix docstring
      
      * Fix test cases
      
      * Fix linting problem
      
      * Skip reduction unittests
      544a2cd1
  23. 23 Sep, 2022 1 commit
  24. 20 Sep, 2022 1 commit
  25. 14 Sep, 2022 1 commit
  26. 05 Sep, 2022 1 commit
  27. 25 Aug, 2022 1 commit
  28. 22 Aug, 2022 1 commit
  29. 18 Aug, 2022 1 commit
  30. 17 Aug, 2022 1 commit
  31. 16 Aug, 2022 1 commit
  32. 29 Jul, 2022 1 commit
    • 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
  33. 26 Jul, 2022 1 commit
  34. 21 Jul, 2022 1 commit
  35. 14 Jul, 2022 1 commit
  36. 07 Jul, 2022 1 commit
  37. 29 Jun, 2022 1 commit