Skip to content

GitLab

Switch branch/tag
  1. 14 Apr, 2023 1 commit
  3. 24 Mar, 2023 1 commit
  5. 20 Mar, 2023 1 commit
  7. 09 Mar, 2023 1 commit
  9. 22 Feb, 2023 1 commit
  11. 17 Feb, 2023 2 commits
  13. 08 Feb, 2023 1 commit
    • ZhenyuLU_Heliodore's avatar
      [Transform] Add SVDPE Transform Module (#5121) · 5598503a
      ZhenyuLU_Heliodore authored 
      * add SVD positional encoding

* modify importing module

* Fixed certain problems

* Change the test unit to a nonsigular one

* Fixed typo and make accord with lintrunner

* added svd_pe into dgl.rst

* Modified dgl.rst
      5598503a
  15. 06 Jan, 2023 1 commit
  17. 01 Dec, 2022 1 commit
  19. 16 Nov, 2022 1 commit
  21. 06 Nov, 2022 1 commit
  23. 27 Oct, 2022 1 commit
  25. 08 Oct, 2022 1 commit
  27. 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
  29. 31 Aug, 2022 1 commit
  31. 27 Jul, 2022 1 commit
  33. 29 Jun, 2022 1 commit
  35. 28 Jun, 2022 1 commit
  37. 24 May, 2022 1 commit
  39. 23 May, 2022 1 commit
  41. 16 May, 2022 1 commit
  43. 07 May, 2022 1 commit
  45. 23 Apr, 2022 1 commit
  47. 30 Mar, 2022 1 commit
  49. 24 Feb, 2022 1 commit
  51. 23 Feb, 2022 1 commit
    • Minjie Wang's avatar
      [NN] Rework RelGraphConv and HGTConv (#3742) · 0227ddfb
      Minjie Wang authored 
      * WIP: TypedLinear and new RelGraphConv

* wip

* further simplify RGCN

* a bunch of tweak for performance; add basic cpu support

* update on segmm

* wip: segment.cu

* new backward kernel works

* fix a bunch of bugs in kernel; leave idx_a for future

* add nn test for typed_linear

* rgcn nn test

* bugfix in corner case; update RGCN README

* doc

* fix cpp lint

* fix lint

* fix ut

* wip: hgtconv; presorted flag for rgcn

* hgt code and ut; WIP: some fix on reorder graph

* better typed linear init

* fix ut

* fix lint; add docstring
      0227ddfb
  53. 17 Feb, 2022 1 commit
  55. 30 Jan, 2022 1 commit
    • Quan (Andy) Gan's avatar
      [Sampling] New sampling pipeline plus asynchronous prefetching (#3665) · 701b4fcc
      Quan (Andy) Gan authored 
      * initial update

* more

* more

* multi-gpu example

* cluster gcn, finalize homogeneous

* more explanation

* fix

* bunch of fixes

* fix

* RGAT example and more fixes

* shadow-gnn sampler and some changes in unit test

* fix

* wth

* more fixes

* remove shadow+node/edge dataloader tests for possible ux changes

* lints

* add legacy dataloading import just in case

* fix

* update pylint for f-strings

* fix

* lint

* lint

* lint again

* cherry-picking commit fa9f494

* oops

* fix

* add sample_neighbors in dist_graph

* fix

* lint

* fix

* fix

* fix

* fix tutorial

* fix

* fix

* fix

* fix warning

* remove debug

* add get_foo_storage apis

* lint
      701b4fcc
  57. 29 Jan, 2022 1 commit
  59. 25 Jan, 2022 1 commit
  61. 17 Jan, 2022 1 commit
  63. 21 Oct, 2021 1 commit
    • Xin Yao's avatar
      [Sampling] Implement dgl.compact_graphs() for the GPU (#3423) · a8c81018
      Xin Yao authored 
      * gpu compact graph template

* cuda compact graph draft

* fix typo

* compact graphs

* pass unit test but fail in training

* example using EdgeDataLoader on the GPU

* refactor cuda_compact_graph and cuda_to_block

* update training scripts

* fix linting

* fix linting

* fix exclude_edges for the GPU

* add --data-cpu & fix copyright
      a8c81018
  65. 12 Oct, 2021 1 commit
  67. 17 Sep, 2021 1 commit
  69. 16 Sep, 2021 1 commit
  71. 13 Jul, 2021 1 commit
  73. 12 Jul, 2021 1 commit
  75. 30 Jun, 2021 1 commit
  77. 18 May, 2021 1 commit