1. 07 Nov, 2022 1 commit
    • czkkkkkk's avatar
      [Sparse] Add sparse matrix C++ implementation (#4773) · 7c059e86
      czkkkkkk authored
      * [Sparse] Add sparse matrix C++ implementation
      
      * Add documentation
      
      * Update
      
      * Minor fix
      
      * Move Python code to dgl/mock_sparse2
      
      * Move headers to include
      
      * lint
      
      * Update
      
      * Add dgl_sparse directory
      
      * Move src code to dgl_sparse
      
      * Add __init__.py in tests to avoid naming conflict
      
      * Add dgl sparse so in Jenkinsfile
      
      * Complete docstring & SparseMatrix basic op
      
      * lint
      
      * Disable win tests
      7c059e86
  2. 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
  3. 04 Nov, 2022 3 commits
  4. 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
  5. 01 Nov, 2022 1 commit
  6. 31 Oct, 2022 1 commit
  7. 29 Oct, 2022 1 commit
    • Quan (Andy) Gan's avatar
      [Sampling] Enable sampling with edge masks in sample_etype_neighbors (#4749) · 2bca4759
      Quan (Andy) Gan authored
      * sample neighbors with masks
      
      * oops
      
      * refactor again
      
      * remove
      
      * remove debug code
      
      * rename macro
      
      * address comments
      
      * more stuff
      
      * remove
      
      * fix
      
      * try fix unit test
      
      * oops
      
      * fix test
      
      * oops
      
      * change name
      
      * rename a lot of stuff
      
      * oops
      
      * ugh
      
      * misc fixes
      
      * lint
      
      * address a lot of comments
      
      * lint
      
      * lint
      
      * fix
      
      * that was silly
      
      * fix
      
      * fix
      
      * fix
      
      * oops
      2bca4759
  8. 27 Oct, 2022 2 commits
  9. 26 Oct, 2022 1 commit
  10. 19 Oct, 2022 1 commit
  11. 18 Oct, 2022 1 commit
  12. 17 Oct, 2022 1 commit
    • Rhett Ying's avatar
      [Dist] Reduce peak memory in DistDGL (#4687) · b1309217
      Rhett Ying authored
      * [Dist] Reduce peak memory in DistDGL: avoid validation, release memory once loaded
      
      * remove orig_id from ndata/edata for partition_graph()
      
      * delete orig_id from ndata/edata in dist part pipeline
      
      * reduce dtype size and format before saving graphs
      
      * fix lint
      
      * ETYPE requires to be int32/64 for CSRSortByTag
      
      * fix test failure
      
      * refine
      b1309217
  13. 13 Oct, 2022 1 commit
  14. 12 Oct, 2022 2 commits
  15. 10 Oct, 2022 2 commits
  16. 08 Oct, 2022 3 commits
  17. 30 Sep, 2022 3 commits
  18. 29 Sep, 2022 3 commits
  19. 28 Sep, 2022 1 commit
  20. 27 Sep, 2022 1 commit
  21. 26 Sep, 2022 3 commits
  22. 24 Sep, 2022 1 commit
  23. 23 Sep, 2022 1 commit
  24. 21 Sep, 2022 1 commit
  25. 20 Sep, 2022 2 commits