Skip to content

GitLab

Switch branch/tag
  1. 20 Dec, 2022 1 commit
  3. 15 Dec, 2022 1 commit
  5. 12 Dec, 2022 2 commits
  7. 09 Dec, 2022 1 commit
  9. 07 Dec, 2022 2 commits
  11. 05 Dec, 2022 2 commits
  13. 01 Dec, 2022 1 commit
  15. 29 Nov, 2022 1 commit
  17. 28 Nov, 2022 1 commit
  19. 27 Nov, 2022 1 commit
  21. 20 Nov, 2022 1 commit
  23. 17 Nov, 2022 1 commit
  25. 16 Nov, 2022 1 commit
  27. 14 Nov, 2022 1 commit
  29. 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
  31. 04 Nov, 2022 1 commit
  33. 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
  35. 31 Oct, 2022 1 commit
  37. 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
  39. 27 Oct, 2022 1 commit
  41. 12 Oct, 2022 1 commit
  43. 10 Oct, 2022 1 commit
  45. 30 Sep, 2022 1 commit
  47. 29 Sep, 2022 2 commits
  49. 28 Sep, 2022 1 commit
  51. 27 Sep, 2022 2 commits
  53. 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
  55. 24 Sep, 2022 1 commit
  57. 23 Sep, 2022 1 commit
  59. 20 Sep, 2022 2 commits
  61. 15 Sep, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Import PyTorch's CUDA stream management (#4503) · 9a00cf19
      Xin Yao authored 
      * add set_stream

* add .record_stream for NDArray and HeteroGraph

* refactor dgl stream Python APIs

* test record_stream

* add unit test for record stream

* use pytorch's stream

* fix lint

* fix cpu build

* address comments

* address comments

* add record stream tests for dgl.graph

* record frames and update dataloder

* add docstring

* update frame

* add backend check for record_stream

* remove CUDAThreadEntry::stream

* record stream for newly created formats

* fix bug

* fix cpp test

* fix None c_void_p to c_handle
      9a00cf19
  63. 14 Sep, 2022 1 commit
  65. 07 Sep, 2022 1 commit
    • ndickson-nvidia's avatar
      [Feature] Added exclude_self and output_batch to knn graph construction... · 684f66b7
      ndickson-nvidia authored 
      
[Feature] Added exclude_self and output_batch to knn graph construction (Issues #4323 #4316) (#4389)

* * Added "exclude_self" and "output_batch" options to knn_graph and segmented_knn_graph
* Updated out-of-date comments on remove_edges and remove_self_loop, since they now preserve batch information

* * Changed defaults on new knn_graph and segmented_knn_graph function parameters, for compatibility; pytorch/test_geometry.py was failing

* * Added test to ensure dgl.remove_self_loop function correctly updates batch information

* * Added new knn_graph and segmented_knn_graph parameters to dgl.nn.KNNGraph and dgl.nn.SegmentedKNNGraph

* * Formatting

* * Oops, I missed the one in segmented_knn_graph when I fixed the similar thing in knn_graph

* * Fixed edge case handling when invalid k specified, since it still needs to be handled consistently for tests to pass
* Fixed context of batch info, since it must match the context of the input position data for remove_self_loop to succeed

* * Fixed batch info resulting from knn_graph when output_batch is true, for case of 3D input tensor, representing multiple segments

* * Added testing of new exclude_self and output_batch parameters on knn_graph and segmented_knn_graph, and their wrappers, KNNGraph and SegmentedKNNGraph, into the test_knn_cuda test

* * Added doc comments for new parameters

* * Added correct handling for uncommon case of k or more coincident points when excluding self edges in knn_graph and segmented_knn_graph
* Added test cases for more than k coincident points

* * Updated doc comments for output_batch parameters for clarity

* * Linter formatting fixes

* * Extracted out common function for test_knn_cpu and test_knn_cuda, to add the new test cases to test_knn_cpu

* * Rewording in doc comments

* * Removed output_batch parameter from knn_graph and segmented_knn_graph, in favour of always setting the batch information, except in knn_graph if x is a 2D tensor
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      684f66b7
  67. 15 Aug, 2022 1 commit