Skip to content

GitLab

Switch branch/tag
  1. 13 Jul, 2021 3 commits
  3. 25 Jun, 2021 1 commit
  5. 24 Jun, 2021 1 commit
  7. 22 Jun, 2021 1 commit
  9. 11 Jun, 2021 2 commits
    • Jinjing Zhou's avatar
      [NN] Add fast path for GateGCNConv when it has only one edge type (#2994) · 2570d412
      Jinjing Zhou authored 
      * fix gatedgcn

* fix lint
      2570d412
    • nv-dlasalle's avatar
      [Feature] Allow using NCCL for communication in dgl.NodeEmbedding and dgl.SparseOptimizer (#2824) · 17d604b5
      nv-dlasalle authored 
      

* Split from NCCL PR

* Fix type in comment

* Expand documentation for sparse_all_to_all_push

* Restore previous behavior in example

* Re-work optimizer to use NCCL based on gradient location

* Allow for running with embedding on CPU but using NCCL for gradient exchange

* Optimize single partition case

* Fix pylint errors

* Add missing include

* fix gradient indexing

* Fix line continuation

* Migrate 'first_step'

* Skip tests without enough GPUs to run NCCL

* Improve empty tensor handling for pytorch 1.5

* Fix indentation

* Allow multiple NCCL communicator to coexist

* Improve handling of empty message

* Update python/dgl/nn/pytorch/sparse_emb.py
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>

* Update python/dgl/nn/pytorch/sparse_emb.py
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>

* Keepy empty tensor dimensionaless

* th.empty -> th.tensor

* Preserve shape for empty non-zero dimension tensors

* Use shared state, when embedding is shared

* Add support for gathering an embedding

* Fix typo

* Fix more typos

* Fix backend call

* Use NodeDataLoader to take advantage of ddp

* Update training script to share memory

* Only squeeze last dimension

* Better handle empty message

* Keep embedding on the target device GPU if dgl_sparse if false in RGCN example

* Fix typo in comment

* Add asserts

* Improve documentation in example
Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      17d604b5
  11. 07 Jun, 2021 1 commit
  13. 01 Jun, 2021 1 commit
  15. 19 May, 2021 1 commit
  17. 07 May, 2021 1 commit
  19. 03 May, 2021 1 commit
  21. 27 Apr, 2021 1 commit
  23. 25 Apr, 2021 1 commit
  25. 09 Apr, 2021 1 commit
  27. 18 Mar, 2021 2 commits
  29. 09 Mar, 2021 1 commit
  31. 03 Mar, 2021 1 commit
  33. 19 Feb, 2021 1 commit
  35. 12 Feb, 2021 1 commit
  37. 29 Jan, 2021 1 commit
  39. 28 Jan, 2021 1 commit
  41. 27 Jan, 2021 1 commit
    • xiang song(charlie.song)'s avatar
      [Feature] Add support for sparse embedding (#2451) · a7e941c3
      xiang song(charlie.song) authored 
      

* Add sparse embedding for dgl and update rgcn example

* upd

* Fix

* Revert "Fix"

This reverts commit 4da87cdfb8b8c3506b7fc7376cd2385ba8045c2a.

* Fix

* upd

* upd

* Fix

* Add unitest and update impl

* fix

* Clean up rgcn example code

* upd

* upd

* update

* Fix

* update score

* sparse for sage

* remove model sparse

* upd

* upd

* remove global norm

* revert delete model_sparse.py

* update according to comments

* Fix doc

* upd

* Fix test

* upd

* lint

* lint

* lint

* upd

* upd

* clean up
Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-56-220.ec2.internal>
      a7e941c3
  43. 26 Jan, 2021 1 commit
  45. 14 Jan, 2021 1 commit
  47. 13 Jan, 2021 1 commit
  49. 08 Jan, 2021 1 commit
  51. 07 Jan, 2021 1 commit
  53. 04 Jan, 2021 1 commit
  55. 07 Dec, 2020 1 commit
  57. 03 Sep, 2020 1 commit
  59. 18 Aug, 2020 1 commit
  61. 17 Aug, 2020 2 commits
  63. 13 Aug, 2020 1 commit
  65. 12 Aug, 2020 2 commits
    • Tianjun Xiao's avatar
      [Doc] NN doc refactor Conv Layers (#1672) · 69f5869f
      Tianjun Xiao authored 
      

* go through gcn, relgcn

* fix tagconv formula

* fix doc in sageconv

* fix sgconv doc

* replace hat with tilde

* more comments on gmmconv

* fix agnnconv chebconv doc

* modify nnconv doc

* remove &

* add nn conv examples

* Rebase master

* More merge conflicts

* check homo

* add back self loop for some convs, check homo in tranform

* add example for denseconv

* add example and doc for dotgat and cfconv

* check in-degree for graphconv

* add language fix

* gconv address all comments

* another round of change based on api template

* change agnn

* go through agnn, appnp, atomic, cf, cheb, dense, gat, sage modules

* finish pytorch part of nn conv

* mxnet graphconv done

* tensorflow graphconv works

* add new modules into doc

* add comments to not split code

* refine doc

* resr

* more comments

* more fix

* finish conv and dense conv part api

* pylint fix

* fix pylink

* fix pylint

* more fix

* fix

* fix test fail because zere in degree

* fix test fail

* sage is not update for mxnet tf
Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-81.us-east-2.compute.internal>
      69f5869f
    • xiang song(charlie.song)'s avatar
      [Feature] Lazy copy ndata, edata to device (#1986) · 5e34ca8b
      xiang song(charlie.song) authored 
      

* Lazy to device

* remove print

* lint

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* lint

* Fix

* Revert "Fix"

This reverts commit 615c9b8f80f5f6ee2ab43c849a22f0083deedf3b.

* Add test for frame lazy update

* disable tensorflow

* upd
Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-51-214.ec2.internal>
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      5e34ca8b
  67. 11 Aug, 2020 1 commit
    • Quan (Andy) Gan's avatar
      [Feature] Edge DataLoader for edge classification & link prediction (#1828) · f5eb80d2
      Quan (Andy) Gan authored 
      * clean commit

* oops forgot the most important files

* use einsum

* copy feature from frontier to block

* Revert "copy feature from frontier to block"

This reverts commit 5224ec963eb6a3ef1b6ab74d8ecbd44e4e42f285.

* temp fix

* unit test

* fix

* revert jtnn

* lint

* fix win64

* docstring fixes and doc indexing

* revert einsum in sparse bidecoder

* fix some examples

* lint

* fix due to some tediousness in remove_edges

* addresses comments

* fix

* more jtnn fixes

* fix
      f5eb80d2