1. 29 Jan, 2021 1 commit
  2. 28 Jan, 2021 6 commits
  3. 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
  4. 26 Jan, 2021 4 commits
  5. 25 Jan, 2021 3 commits
    • Zihao Ye's avatar
    • Zihao Ye's avatar
      [feature] Implement missing CUDA operators for COO format (part 1). (#2565) · 0f9056ed
      Zihao Ye authored
      
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * fix
      
      * upd
      
      * upd
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      0f9056ed
    • Da Zheng's avatar
      [Distributed] Heterogeneous graph support (#2457) · 25ac3344
      Da Zheng authored
      
      
      * Distributed heterograph (#3)
      
      * heterogeneous graph partition.
      
      * fix graph partition book for heterograph.
      
      * load heterograph partitions.
      
      * update DistGraphServer to support heterograph.
      
      * make DistGraph runnable for heterograph.
      
      * partition a graph and store parts with homogeneous graph structure.
      
      * update DistGraph server&client to use homogeneous graph.
      
      * shuffle node Ids based on node types.
      
      * load mag in heterograph.
      
      * fix per-node-type mapping.
      
      * balance node types.
      
      * fix for homogeneous graph
      
      * store etype for now.
      
      * fix data name.
      
      * fix a bug in example.
      
      * add profiler in rgcn.
      
      * heterogeneous RGCN.
      
      * map homogeneous node ids to hetero node ids.
      
      * fix graph partition book.
      
      * fix DistGraph.
      
      * shuffle eids.
      
      * verify eids and their mappings when loading a partition.
      
      * Id map from homogneous Ids to per-type Ids.
      
      * verify partitioned results.
      
      * add test for distributed sampler.
      
      * add mapping from per-type Ids to homogeneous Ids.
      
      * update example.
      
      * fix DistGraph.
      
      * Revert "add profiler in rgcn."
      
      This reverts commit 36daaed8b660933dac8f61a39faec3da2467d676.
      
      * add tests for homogeneous graphs.
      
      * fix a bug.
      
      * fix test.
      
      * fix for one partition.
      
      * fix for standalone training and evaluation.
      
      * small fix.
      
      * fix two bugs.
      
      * initialize projection matrix.
      
      * small fix on RGCN.
      
      * Fix rgcn performance (#17)
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix.
      
      * fix test.
      
      * fix lint.
      
      * test partitions.
      
      * remove redundant test for partitioning.
      
      * remove commented code.
      
      * fix partition.
      
      * fix tests.
      
      * fix RGCN.
      
      * fix test.
      
      * fix test.
      
      * fix test.
      
      * fix.
      
      * fix a bug.
      
      * update dmlc-core.
      
      * fix.
      
      * fix rgcn.
      
      * update readme.
      
      * add comments.
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      
      * fix.
      
      * fix.
      
      * add div_int.
      
      * fix.
      
      * fix.
      
      * fix lint.
      
      * fix.
      
      * fix.
      
      * fix.
      
      * adjust.
      
      * move code.
      
      * handle heterograph.
      
      * return pytorch tensor in GPB.
      
      * remove some tests in example.
      
      * add to_block for distributed training.
      
      * use distributed to_block.
      
      * remove unnecessary function in DistGraph.
      
      * remove distributed to_block.
      
      * use pytorch tensor.
      
      * fix a bug in ntypes and etypes.
      
      * enable norm.
      
      * make the data loader compatible with the old format.
      
      * fix.
      
      * add comments.
      
      * fix a bug.
      
      * add test for heterograph.
      
      * support partition without reshuffle.
      
      * add test.
      
      * support partition without reshuffle.
      
      * fix.
      
      * add test.
      
      * fix bugs.
      
      * fix lint.
      
      * fix dataset.
      
      * fix for mxnet.
      
      * update docstring.
      
      * rename to floor_div
      
      * avoid exposing NodePartitionPolicy and EdgePartitionPolicy.
      
      * fix docstring.
      
      * fix error.
      
      * fixes.
      
      * fix comments.
      
      * rename.
      
      * rename.
      
      * explain IdMap.
      
      * fix docstring.
      
      * fix docstring.
      
      * update docstring.
      
      * remove the code of returning heterograph.
      
      * remove argument.
      
      * fix example.
      
      * make GraphPartitionBook an abstract class.
      
      * fix.
      
      * fix.
      
      * fix a bug.
      
      * fix a bug in example
      
      * fix a bug
      
      * reverse heterograph sampling.
      
      * temp fix.
      
      * fix lint.
      
      * Revert "temp fix."
      
      This reverts commit c450717b9f578b8c48769c675f2a19d6c1e64381.
      
      * compute norm.
      
      * Revert "reverse heterograph sampling."
      
      This reverts commit bd6deb7f52998de76508f800441ff518e2fadcb9.
      
      * fix.
      
      * move id_map.py
      
      * remove check
      
      * add more comments.
      
      * update docstring.
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-62-171.ec2.internal>
      25ac3344
  6. 24 Jan, 2021 3 commits
  7. 21 Jan, 2021 2 commits
  8. 20 Jan, 2021 3 commits
    • xnouhz's avatar
      [Feature] QM9 Dataset Support (#2521) · 2c6d0716
      xnouhz authored
      
      
      * [Model] MixHop for Node Classification task
      
      * [docs] update
      
      * [docs] update
      
      * [fix] remove seed option
      
      * [fix] update readme
      
      * [feature] support qm9 dataset
      
      * [style] update
      
      * [docs] fix the details
      
      * [fix] indexing only support int
      
      * [style] update
      
      * [fix] multiple backends support
      
      * [docs] add qm9
      
      * [fix] Z type: float32 -> int32
      
      * [fix] Z type: int32 -> long
      
      * [docs] add ref
      
      * [docs] fix
      
      * [docs] update
      
      * [docs] update
      
      * [fix] test eval
      
      * [docs] fix example
      Co-authored-by: default avatarxnuohz@126.com <ubuntu@ip-172-31-44-184.us-east-2.compute.internal>
      Co-authored-by: default avatarMufei Li <mufeili1996@gmail.com>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      2c6d0716
    • Jinjing Zhou's avatar
      [Test] API benchmarks (#2542) · 0c156573
      Jinjing Zhou authored
      
      
      * add bench jenkins
      
      * instance type
      
      * fix
      
      * fix
      
      * fix
      
      * 111
      
      * test
      
      * 111
      
      * 111
      
      * fix
      
      * test
      
      * run
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * publish results
      
      * 111
      
      * regression
      
      * launch ec2 script
      
      * fix
      
      * add
      
      * run on master
      
      * change
      
      * rrr
      
      * run gpu
      
      * fix
      
      * fix
      
      * try fix
      
      * fix
      
      * ff
      
      * fix
      
      * fix
      
      * fix
      
      * refactor
      
      * fix
      
      * fix
      
      * update
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * remove import torchtext
      
      * add shm size
      
      * update
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix this!!!!
      
      * 111
      
      * fix
      
      * remove verbose
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * update readme
      
      * fix
      
      * fix
      
      * fix
      
      * change asv default to head
      
      * commit sage and rgcn
      
      * fix
      
      * update
      
      * add benchmarks
      
      * add
      
      * fix
      
      * update
      
      * remove RandomState
      
      * tmp remove
      
      * new batch
      
      * fix
      
      * fix
      
      * fix
      
      * address comment
      
      * fix warning
      
      * fix
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      0c156573
    • Josh Carty's avatar
      Add a TensorFlow example for SGC (#2480) · 8a2b54d1
      Josh Carty authored
      
      
      This is a TensorFlow implementation of Simple Graph Convolution. For
      simplicity, we use the implementation of AdamW in TensorFlow Addons
      that differs from Adam in PyTorch. This results in worse performance
      on the evaluation datasets.
      Co-authored-by: default avatarJosh Carty <joshuac@spotify.com>
      Co-authored-by: default avatarZihao Ye <expye@outlook.com>
      8a2b54d1
  9. 19 Jan, 2021 2 commits
  10. 18 Jan, 2021 3 commits
    • 张天启's avatar
      [Example] Add HGP-SL example for pytorch backend (#2515) · b36b6c26
      张天启 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'
      Co-authored-by: default avatarzhangtianqi <tianqizh@amazon.com>
      b36b6c26
    • Minjie Wang's avatar
      Add one paper · 1caf01d0
      Minjie Wang authored
      1caf01d0
    • Minjie Wang's avatar
      Update README.md · 6b463820
      Minjie Wang authored
      6b463820
  11. 17 Jan, 2021 1 commit
  12. 15 Jan, 2021 2 commits
  13. 14 Jan, 2021 4 commits
    • Mufei Li's avatar
      [Feature] Add Device Flag in Data Loaders (#2450) · 5da3439c
      Mufei Li authored
      
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      5da3439c
    • Minjie Wang's avatar
      Update example README.md · b1fb3c15
      Minjie Wang authored
      b1fb3c15
    • Andrew's avatar
      Update CONTRIBUTORS.md (#2538) · 0ab9167c
      Andrew authored
      0ab9167c
    • Jinjing Zhou's avatar
      Add API benchmarks (#2522) · c29daaef
      Jinjing Zhou authored
      
      
      * add bench jenkins
      
      * instance type
      
      * fix
      
      * fix
      
      * fix
      
      * 111
      
      * test
      
      * 111
      
      * 111
      
      * fix
      
      * test
      
      * run
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * publish results
      
      * 111
      
      * regression
      
      * launch ec2 script
      
      * fix
      
      * add
      
      * run on master
      
      * change
      
      * rrr
      
      * run gpu
      
      * fix
      
      * fix
      
      * try fix
      
      * fix
      
      * ff
      
      * fix
      
      * fix
      
      * fix
      
      * refactor
      
      * fix
      
      * fix
      
      * update
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * remove import torchtext
      
      * add shm size
      
      * update
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix this!!!!
      
      * 111
      
      * fix
      
      * remove verbose
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * update readme
      
      * fix
      
      * fix
      
      * fix
      
      * change asv default to head
      
      * commit sage and rgcn
      
      * fix
      
      * update
      
      * add benchmarks
      
      * add
      
      * fix
      
      * update
      
      * remove RandomState
      
      * tmp remove
      Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
      c29daaef