提交

提交图

3215 次代码提交

作者 SHA1 备注 提交日期
Tingzhang Zhao 4579bbf7b7 Update README.md (#2853)
Change the format of  README.md
2021-04-21 13:05:25 +08:00
Quan (Andy) Gan 6232ee8d44 Update README.md (#2820)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-04-20 13:17:30 +08:00
Tingzhang Zhao e58eeebfb6 [Model] Add RECT example (#2813)
* Update README.md

Add description of RECT

* [Example] Add implementation of RECT

[Example] Add implementation of RECT

* Update classify.py

Modify the class names and the function names mentioned above

* Update main.py

Modify the function names mentioned above

* Update label_utils.py

Adjust the comments

* Update README.md

Add the github information

Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
2021-04-19 21:48:38 +08:00
nv-dlasalle bbebde4660 [Performance] Track sorted status of COO from creation (#2645)
* Add row/col sorted flags

* improve sorting paths

* Remove print statement

* Keep track of sorted matrices

* Remove sort check in to_block

* Improve CPU sorted COO->CSR

* Handle the zero edge case

* Remove omp default clause to work with MSVC

* Update comments on sorted COO->CSR cpu implementatoin

* Expose sorted to python interface

* Make check_sorted default to false for dgl.graph()

* remove check sorted; add utests

* remove check_sorted flag

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-16 10:30:07 +08:00
Jinjing Zhou 302125efd4 [Test] Add comment trigger for regression test (#2827)
* add

* fix

* f

* 111

* print

* fix

* ci

* f

* fix regex

* fix

* t

* fix

* add

* test

* fix

* fix

* fff

* fix

* 1113

* v2

* 222

* ttt2

* f2

* ff

* 11

* 121

* try

* add

* 222

* asd

* t2

* fix

* install dependency

* ff

* 图

* fix

* fix

* fix

* fix

* escape

* fix

* add url

* fix branches

* add url

* fix

* fix

* fix

* to csv

* update link

* fix

* fix

* fix
2021-04-15 16:10:29 +08:00
nv-dlasalle e70138bbe4 [Performance][GPU] Enable GPU uniform edge sampling (#2716)
* Start on uniform GPU sampling

* Save more work

* Get cu file compiling

* Update sampling

* More changes

* Get GPU sampling for uniform probabilities solved

* Fix batch tensor migration

* Fix

* update kernels

* expand blocking

* Undo testing change

* Cut down on sampling overhead

* Fix replacement

* Update unit tests

* Add option to gpu sample in graphsage

* Copy only csc to gpu

* Add ogbn support

* Fix linting

* Remove nvtx from sample

* Improve documentation and error checking

* Expand documentation

* Update assert checking

* delete extra space

* Use standard dataloader when dataset is a dictionary

* ogb -> ogbn

* Fix edge selection determinism

* Fix typos

* Remove nvtx

* Add comment for self.fanout_arrays and assert

* Fix linting

* Migrate to scalarbatcher

* Fix indentation

* Fix batcher

* Fix indexing

* Only use databatcher for GPU

* Convert to DGL NDArray to PyTorch Tensor

* Add optimization for PyTorch's F.tensor() for list of GPU tensors

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-04-14 18:55:09 -07:00
maqy 195f99362d [example] multi gpus training for mag240m (#2835)
* multi gpus training for mag240m

* remove unused function

* small changes

* more fix

* use multi gpus to eval and more fix

Co-authored-by: Quan Gan <coin2028@hotmail.com>
2021-04-13 20:29:43 +08:00
Da Zheng de5e8e2392 [Distributed] Fix a bug for graphs without node/edge data. (#2838)
* fix.

* test distributed graph without node/edge data.

* remove some tests.

* fix lint
2021-04-12 22:35:01 -07:00
Chen Sirui afc83aa216 Graphsim (#2794)
* Add hgat example

* Add experiment

* Clean code

* clear the code

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Change the code title and folder name

* Ready to merge

* Prepare for rebase and change message passing function

* use git ignore to handle empty file

* change file permission to resolve empty file

* Change permission

* change file mode

* Finish Coding

* working code cpu

* pyg compare

* Accelerate with batching

* FastMode Enabled

* update readme

* Update README.md

* refractor code

* add graphsim code

* modified code

* few fix

* Modified graphsim

* Simple Model Added

* Clean up code

* Refractor the code for Merge

* Bugfix enable gradient when train

* update readme and format

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-45-47.ap-northeast-1.compute.internal>
2021-04-12 19:02:36 +08:00
Quan (Andy) Gan 4c7476c812 [Doc] update readme (#2737)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-12 17:02:35 +08:00
Tong He 5164641746 [Doc] add 4 papers to awesome (#2836) 2021-04-12 17:01:18 +08:00
TianleZhang d70eade4f2 bug fix type mismatch (#2830)
int has no attribute of item

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-04-11 20:33:03 +08:00
Tianqi Zhang (张天启) e83d0a8007 [Feature] Add kd-tree implementation (CPU) for kNN (#2767)
* add submodule nanoflann

* finish python API for knn

* finish ndarray adaptor

* finish cpu-kdtree version of knn

* use openmp

* add endline

* upt

* upt

* fix format and code style

* upt

* add warning for gpu-cpu copy

* avoid contiguous copy

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-04-09 17:39:34 +08:00
Tonny-Gu 0fce09074d [Distributed] Bind to the correct NIC (#2798)
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-04-08 17:56:30 -07:00
Da Zheng bfbbefa793 [Distributed] Fix a bug in multiprocessing sampling. (#2826)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-73-81.ec2.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-04-08 01:29:33 -07:00
Quan (Andy) Gan 48b9ecd201 Fix discrepancy in dgl.dataloading.EdgeDataLoader (#2819)
* fix discrepancy

* address comments
2021-04-07 16:08:14 +08:00
xnouhz 2afa3598ba [Example] JKNet (#2795)
* [example] jknet

* update

* update

* update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-04-07 14:29:33 +08:00
Ruoyun Huang 2980137719 [Doc] Typo in command example. (#2816)
* Typo in command example. 

process_nowplaying_rs.py is the right name under this folder.

* Update README.md

* Update README.md

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-04-07 09:43:37 +08:00
Sergey Ivanov 672e322719 [Model] Add BGNN example (#2740)
* Add BGNN example

* Update bgnn example

* Remove uncessary files

* update model index

* style fix

* Add Readme for BGNN example

* Update documentation; remove preprocessing

Co-authored-by: s.ivanov <s.ivanov@mesos-slave080-am6.central.criteo.prod>
Co-authored-by: s.ivanov <s.ivanov@mesos-slave150-am6.central.criteo.prod>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
Co-authored-by: s.ivanov <s.ivanov@mesos-slave148-am6.central.criteo.prod>
Co-authored-by: s.ivanov <s.ivanov@mesos-slave078-am6.central.criteo.prod>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-07 00:16:09 +08:00
Quan (Andy) Gan 3301bd0bcc bump nightly to 0.7 (#2814) 2021-04-06 13:51:26 +08:00
KounianhuaDu e61138be19 [Example] CompGCN (#2768)
* compgcn

* readme

* readme

* update

* readme

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-04-05 20:45:38 +08:00
Quan (Andy) Gan cfe6e70b7a RGAT baseline for OGB-LSC node classification challenge (#2810)
* RGAT baseline

* update links (data.dgl.ai hasn't refreshed yet)

* nvm still use dgl.ai domain

* descriptions

* actually it took less

* address comments

* stop worrying about cache

* update links

* oops
2021-04-05 18:21:18 +08:00
Quan (Andy) Gan 05c53ca38e [Performance] Prefer parallelized conversion to CSC from COO instead of transposing CSR (#2793)
* fix coo2csr speed

* add comments
2021-04-05 16:38:36 +08:00
Da Zheng 86229d426f [Distributed] Automatically setting the number of OMP threads for trainers (#2812)
* set omp thread.

* add comment.

* fix.
2021-04-04 11:50:28 -07:00
Yu Sun cba5af22eb [bugfix] Fix bugs in vgae (#2727)
* [Example]Variational Graph Auto-Encoders

* change dgl dataset to single directional graph

* clean code

* refresh

* fix bug

* fix bug

* fix bug

* add gpu

Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-03 20:02:04 +08:00
pawelpiotrowicz b2e35e6aa9 [Performance] Linear UniformChoice optimization (#2710)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-04-01 16:29:49 +08:00
Minjie Wang 54c74803b2 [CI] Disable distributed KVStore UT (#2805) 2021-04-01 16:00:56 +08:00
Xiangkun Hu d6f856318a [Doc] Add two papers to Awesome DGL Papers (#2799)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* data pipeline user guide

* remove chapter numbers

* Update data.rst

* image in dataset userguide

* update links using ref

* modify the link of save_graphs and load_graphs in dataset user guide

* move image to s3 server.

* fix links and ref

* Hide implementations that may cause confusions to the user

* add papers

* fix number of classes in Amazon Cobuy datasets

* add two papers to Awesome paper list

* Update README.md

* add four papers

* add two papers
2021-04-01 14:38:28 +08:00
Xiangkun Hu b49cd0952f [Bugfix] Fix feature type of 'ENZYMES' in TUDataset (#2800)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets

* add four papers

* fix feature type of TU dataset

* fix readme

* remove commented codes

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-01 14:36:55 +08:00
milesial 1f4c0b7193 Fix processing of QM9EdgeDataset (#2801) 2021-04-01 10:13:51 +08:00
Da Zheng e36c5db614 [Distributed] Simplify distributed API (#2775)
* remove num_workers.

* remove num_workers.

* remove num_workers.

* remove num-servers.

* update error message.

* update docstring.

* fix docs.

* fix tests.

* fix test.

* fix.

* print messages in test.

* fix.

* fix test.

* fix.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
2021-03-29 19:19:23 -07:00
Da Zheng 97863ab85b [Distributed] Cleanup the process pool correctly when the process exits. (#2781)
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-03-29 14:53:51 -07:00
Quan (Andy) Gan 66f7fe8bce Update model.py (#2796) 2021-03-29 16:30:32 +08:00
Mufei Li 2952f3c473 Update subgraph.py (#2784)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-03-29 14:58:39 +08:00
Mufei Li a655123b4e [Example] Baselines for OGB-LSC-PCQM4M (#2778)
* Update

* Add files via upload

* Add files via upload

* Add files via upload
2021-03-29 14:07:07 +08:00
Jinjing Zhou 0576a33b7f Update convert.py (#2789)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-03-28 20:59:52 +08:00
colten a1b5b42af4 remove unused and undefined function (#2791)
Co-authored-by: messi-k <messi.k@kakaocorp.com>
2021-03-28 00:17:11 +08:00
Konstantin Sofeikov e22087aa36 [BUG] fix for zero degree nodes (#2786)
* fix for zero degree nodes

* Update python/dgl/ops/spmm.py

linting fix

Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>

Co-authored-by: Konstantin Sofeikov <konstantin.sofeikov@complyadvantage.com>
Co-authored-by: Zihao Ye <expye@outlook.com>
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
2021-03-25 22:09:53 +08:00
Quan (Andy) Gan 0b57ce1867 [Bug] Disable cpu fp16 (#2783)
* disable cpu fp16

* spell mistakes
2021-03-25 18:04:39 +08:00
kyawlinoo 03482f0a4c [NN] tf nn for edgeConv (#2741)
* tf nn for edgeConv

* Auto stash before merge of "tf_working" and "origin/tf_working"

* clean up

* added test for edge_conv

* fix

* fix

* fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2021-03-25 13:44:07 +08:00
Quan (Andy) Gan 929d863447 [Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753)
* test

* more stuff

* add test

* fixes

* optimize algo

* replace unordered_map with arrays

* lint

* lint x2

* oops

* disable gpu csrmm tests

* remove gpu invocation

* optimize with openmp

* remove python functions

* add back with docstrings

* lint

* lint

* update python interface

* functionize

* functionize

* lint

* lint
2021-03-24 17:24:41 +08:00
Chen Sirui d04d59eef9 [Example] Temporal Graph Neural Network (#2636)
* Add hgat example

* Add experiment

* Clean code

* clear the code

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Add index in README

* Change the code title and folder name

* Ready to merge

* Prepare for rebase and change message passing function

* use git ignore to handle empty file

* change file permission to resolve empty file

* Change permission

* change file mode

* Finish Coding

* working code cpu

* pyg compare

* Accelerate with batching

* FastMode Enabled

* update readme

* Update README.md

* refractor code

* Fix Bug

* add a simple temporal sampling method

* test results

* add train speed result

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* fix bug

* Fixed Grammar and Format Issue

Co-authored-by: Chen <chesirui@3c22fbe5458c.ant.amazon.com>
Co-authored-by: Tianjun Xiao <xiaotj1990327@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-4-63.ap-northeast-1.compute.internal>
Co-authored-by: WangXuhongCN <wangxuhongcn@163.com>
Co-authored-by: WangXuhongCN <34096632+WangXuhongCN@users.noreply.github.com>
2021-03-23 08:57:21 +08:00
Tianqi Zhang (张天启) 6999f88f3f [Model] Add random_start option for Farthest Point Sampler (#2755)
* add random_start option for FPS

* change doc

* change from random_start to start_idx

* change error condition

* change error msg

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-03-22 21:53:46 +08:00
Hengrui Zhang bcffdb82c9 [Example] Add implementation of mvgrl (#2739)
* [Example ]add mvgrl

* [Doc] add mvgrl to readme

* add more comments

* fix typos

* replace tab with space

* [doc] replace tab with space

* [Doc] fix a typo

* fix minor typos

* fix typos

* fix typos

* fix typos

* fix typos

* fix

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-22 21:20:21 +08:00
Kaiqiang Xu e6f6c2eb31 [Bugfix] Update deprecated method name in load_graph.py (#2769)
Method `dataset.num_labels` has been deprecated and replaced by `dataset.num_classes`.  
Updating the method name to avoid runtime warning.

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-03-22 20:07:32 +08:00
nv-dlasalle 0ff7127a0f [Bugfix] Wrap cub with CUB_NS_PREFIX and remove dependency on Thrust to linking issues with Torch 1.8 (#2758)
* Wrap cub with prefixes and remove thrust

* Using counting iterator

Co-authored-by: Zihao Ye <expye@outlook.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-03-22 20:06:57 +08:00
KounianhuaDu 9aac93ff21 Gnnexplainer (#2717)
* gnn-explainer

* gnn-explainer

* gnn-explainer

* gnn-explainer

* fix

* fix

* fix

* readme

* readme

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-03-22 18:13:08 +08:00
Hengrui Zhang 1789972c4b [Bugfix] add default value for BlockSampler (#2771)
* [Bugfix] add default value for BlockSampler

* [Doc] modify user_guide description about MultiLayerDropoutSample

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-03-22 14:02:27 +08:00
Da Zheng 17f863569e fix (#2744) 2021-03-21 22:38:55 -07:00
Da Zheng 74c38a1f58 Print error messages when using TCP socket (#2763)
* print error messages.

* fix.
2021-03-21 21:00:22 -07:00