Skip to content

GitLab

Switch branch/tag
  1. 28 Nov, 2022 2 commits
  3. 27 Nov, 2022 1 commit
  5. 25 Nov, 2022 1 commit
  7. 21 Nov, 2022 1 commit
  9. 20 Nov, 2022 1 commit
  11. 18 Nov, 2022 2 commits
    • kylasa's avatar
      [Dist] Flexible pipeline - Initial commit (#4733) · c8ea9fa4
      kylasa authored 
      * Flexible pipeline - Initial commit

1. Implementation of flexible pipeline feature.
2. With this implementation, the pipeline now supports multiple partitions per process. And also assumes that num_partitions is always a multiple of num_processes.

* Update test_dist_part.py

* Code changes to address review comments

* Code refactoring of exchange_features function into two functions for better readability

* Upadting test_dist_part to fix merge issues with the master branch

* corrected variable names...

* Fixed code refactoring issues.

* Provide missing function arguments to exchange_feature function

* Providing the missing function argument to fix error.

* Provide missing function argument to 'get_shuffle_nids' function.

* Repositioned a variable within its scope.

* Removed tab space which is causing the indentation problem

* Fix issue with the CI test framework, which is the root cause for the failure of the CI tests.

1. Now we read files specific to the partition-id and store this data separately, identified by the local_part_id, in the local process.
2. Similarly as above, we also differentiate the node and edge features type_ids with the same keys as above.
3. These above two changes will help up to get the appropriate feature data during the feature exchange and send to the destination process correctly.

* Correct the parametrization for the CI unit test cases.

* Addressing Rui's code review comments.

* Addressing code review comments.
      c8ea9fa4
    • Tingyu Wang's avatar
      [Model] Add `dgl.nn.CuGraphRelGraphConv` model (#4826) · 4ecd2c85
      Tingyu Wang authored 
      

* added relgraphconv

* add tests for CuGraphRelGraphConv

* move test to tests directory

* update docs

* address PR review 1

* address PR review, continued

* remove stashing ctx.device

* make fanout optional

* fanout -> max_in_degree

* improve docstrings consistency

* initialize parameters directly in __init__

* get rid of shared memory size check

* improve docs

* format comments

* skip test case for regularizer=None
Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
      4ecd2c85
  13. 17 Nov, 2022 2 commits
  15. 16 Nov, 2022 2 commits
  17. 15 Nov, 2022 1 commit
  19. 14 Nov, 2022 1 commit
  21. 12 Nov, 2022 1 commit
  23. 10 Nov, 2022 1 commit
  25. 09 Nov, 2022 1 commit
  27. 07 Nov, 2022 4 commits
  29. 06 Nov, 2022 3 commits
  31. 04 Nov, 2022 3 commits
  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. 01 Nov, 2022 2 commits
  37. 31 Oct, 2022 1 commit
  39. 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
  41. 28 Oct, 2022 1 commit
  43. 27 Oct, 2022 3 commits
  45. 26 Oct, 2022 1 commit
  47. 19 Oct, 2022 3 commits