Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "1051ca81a60073702320b20eb633b178c3dd1c9b"
Switch branch/tag
  1. 24 Sep, 2022 1 commit
  3. 23 Sep, 2022 4 commits
    • kylasa's avatar
      Garbage Collection and memory snapshot code for debugging partitioning... · ace76327
      kylasa authored 
      
 Garbage Collection and memory snapshot code for debugging partitioning pipeline (target as master branch) (#4598)

* Squashed commit of the following:

commit e605a550b3783dd5f24eb39b6873a2e0e79be9c7
Author: kylasa <kylasa@gmail.com>
Date:   Thu Sep 15 14:45:39 2022 -0700

    Delete pyproject.toml

commit f2db9e700d817212b67b5227f6472d218f0c74f2
Author: kylasa <kylasa@gmail.com>
Date:   Thu Sep 15 14:44:40 2022 -0700

    Changes suggested by isort program to sort imports.

commit 5a6078beac6218a4f1fb378c169f04dda7396425
Author: kylasa <kylasa@gmail.com>
Date:   Thu Sep 15 14:39:50 2022 -0700

    addressing code review comments from the CI process.

commit c8e92decb7aebeb32c7467108e16f058491443ab
Author: kylasa <kylasa@gmail.com>
Date:   Wed Sep 14 18:23:59 2022 -0700

    Corrected a typo in the import statement

commit 14ddb0e9b553d5be3ed2c50d82dee671e84ad8c9
Author: kylasa <kylasa@gmail.com>
Date:   Tue Sep 13 18:47:34 2022 -0700

    Memory snapshot code for debugging memory footprint of the graph partitioning pipeline

Squashed commit done

* Addressing code review comments.

* Update utils.py

* dummy change to trigger CI tests
Co-authored-by: default avatarRhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
      ace76327
    • Chang Liu's avatar
      [Example][Refactor] Graphsage link prediction example refactor (#4526) · 533afa85
      Chang Liu authored 
      * Refactor link pred example for graphsage

* Use ogb evaluator + README update

* Update

* Add comments
      533afa85
    • czkkkkkk's avatar
      [Sparse] Add Transposition for SparseMatrix and DiagMatrix (#4597) · db64cc37
      czkkkkkk authored 
      

* [Sparse] Add Transposition

* Fix docstring

* Fix linting problem

* Minor fix

* Minor fix
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      db64cc37
    • Vibhu Jawa's avatar
      [Bugfix][Hotfix] Fix cugraph Pytorch CI failures (#4622) · d04534c9
      Vibhu Jawa authored 
      

* get_more_logs

* Update cugraph_unit_test.sh

* Update test_basics.py

* Update cugraph_unit_test.sh

* Update tests/cugraph/test_basics.py
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      d04534c9
  5. 22 Sep, 2022 4 commits
  7. 21 Sep, 2022 2 commits
  9. 20 Sep, 2022 4 commits
  11. 19 Sep, 2022 3 commits
  13. 16 Sep, 2022 14 commits
  15. 15 Sep, 2022 3 commits
  17. 14 Sep, 2022 2 commits
  19. 08 Sep, 2022 1 commit
  21. 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
  23. 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