Skip to content

GitLab

Switch branch/tag
  1. 08 Feb, 2021 2 commits
    • nv-dlasalle's avatar
      [Sampling] Implement `dgl.to_block()` for the GPU (#2339) · bc3a532f
      nv-dlasalle authored 
      

* Add start of to_block gpu implementation

* Pull in more changes from 0.4.2 cuda_to_block

* Move more code to IdArray

* Refactor DeviceNodeMapMaker

* Updates

* get compiling

* Integrate to_block

* Fix ID allocation

* Minor fixes

* Cleanup cuda calls to use cuda_common

* Reduce kernel calls

* Lint cleanup

* Expand documentation

* Remove unused function

* Rename variables for consistency

* Add doxygen comments

* Fix file extension

* Remove raw asynccopy for deviceapi

* Remove unused function

* Fix block/tile configuration

* Add cuda_device_common.cuh

* Add basic hashtable

* Migrate part of hashtable

* Refactor to use external hashtable

* Make functions members

* Format hash table functions

* Migrate duplicate filling

* Move last function over

* Refactor with cu file

* lint c++ code

* Move context check to C++ code

* Use macro switch

* Add missing files

* Update docstring

* update docs

* Move atomic functions

* Refactor hashtable

* Fix linting

* Expand docs

* Fix mismatched argument names

* Switch doxygen comments from using @param to \param
Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      bc3a532f
    • Tianjun Xiao's avatar
      [Example]Add dependencies for GNN-Film and dagnn (#2633) · ae18d7f0
      Tianjun Xiao authored 
      * add dependencies for GNN-Film and dagnn

* fix

* fix

* fix
      ae18d7f0
  3. 07 Feb, 2021 1 commit
  5. 05 Feb, 2021 5 commits
  7. 04 Feb, 2021 3 commits
  9. 03 Feb, 2021 5 commits
  11. 02 Feb, 2021 3 commits
  13. 31 Jan, 2021 1 commit
  15. 29 Jan, 2021 2 commits
  17. 28 Jan, 2021 6 commits
  19. 27 Jan, 2021 6 commits
    • 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
    • Minjie Wang's avatar
      [Test] Performance benchmarks for DGL kernels (#2582) · 362f72cb
      Minjie Wang authored 
      * add initial kernel benchmarks

* finished kernel benchmarks

* add desc
      362f72cb
    • Tong He's avatar
      remove hash_func from dataset object (#2572) · 12f64296
      Tong He authored 
      
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      12f64296
    • Tianqi Zhang (张天启)'s avatar
      [Example] Add Graph Cross Net (GXN) example for pytorch backend (#2559) · 8e981db9
      Tianqi Zhang (张天启) authored 
      

* add sagpool example for pytorch backend

* polish sagpool example for pytorch backend

* [Example] SAGPool: use std variance

* [Example] SAGPool: change to std

* add sagpool example to index page

* add graph property prediction tag to sagpool

* [Example] add graph classification example HGP-SL

* [Example] fix sagpool

* fix bug

* [Example] change tab to space in README of hgp-sl

* remove redundant files

* remote redundant network

* [Example]: change link from code to doc in HGP-SL

* [Example] in HGP-SL, change to meaningful name

* [Example] Fix path mistake for 'hardgat'

* [Bug Fix] Fix undefined var bug in LegacyTUDataset

* upt

* [Bug Fix] Fix cache file name bug in TUDataset

* [Example] Add GXN example for pytorch backend

* modify readme

* add more exp result
Co-authored-by: default avatarzhangtianqi <tianqizh@amazon.com>
Co-authored-by: default avatarTong He <hetong007@gmail.com>
      8e981db9
    • Minjie Wang's avatar
      [Doc] Fix docstring of aten::Sort (#2579) · 9fc5eed6
      Minjie Wang authored 
      A small patch to #2391.
      9fc5eed6
    • nv-dlasalle's avatar
      [Performance] Improve COO to CSR, and sort columns of CSR only when necessary. (#2391) · 2576647c
      nv-dlasalle authored 
      * Remove double-checking sorted

* Remove sorting of CSR by default

* Update unit test to use unsorted matix

* delete whitespace

* Expand unit tests

* Replace cusparse sort

* Fix row column sorting

* Explicitly don't sort columns

* Fix linting errors

* Fix bit-width calculation

* Fix sorting assertion and unit test

* Fix linting

* Improve CPU COO2CSR

* Remove references

* Rename and add documentation to edge encoding/decoding funcionts

* Fix sorting keys as 64 bit

* Revert cosmetic changes to unit tests

* Update documentation

* Update complexity documentation for coo to csr conversion

* Remove COOIsSorted check in CPU implementation too
      2576647c
  21. 26 Jan, 2021 4 commits
  23. 25 Jan, 2021 2 commits