提交

2212 次代码提交

作者 SHA1 备注 提交日期
Quan Gan 1453bf2ceb update version 0.8.1 2022-04-15 16:37:36 +00:00
Quan (Andy) Gan 4375c2d7e4 [Doc] Fix documentation in dgl.multiprocessing namespace (#3929)
* fix docs

* remove

* oh

* fix
2022-04-16 00:35:06 +08:00
WangYQ f5bba284f8 [Example] Add EGES example (#3756)
* add eges example

* remove csv files and add data link

* Update README.md

* Update main.py

* Update model.py

* Update sampler.py

* Update utils.py

* Update model.py

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-14 16:23:08 +08:00
Tong He a0bf5daa65 fix (#3903)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-04-13 12:28:19 +08:00
Mufei Li c380f0b5fa Update gatv2.py (#3921)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-13 09:47:52 +08:00
Quan (Andy) Gan e06e63d5d5 [Feature] Enable UVA sampling with CPU indices (#3892)
* enable UVA sampling with CPU indices

* add docs

* add more docs

* lint

* fix

* fix

* better error message

* use mp.Barrier instead of queues

* revert

* revert

* oops

* revert dgl.multiprocessing.spawn

* Update pytorch.py
2022-04-12 20:56:51 +08:00
Quan (Andy) Gan 0d878ff8ef [Example] Cleaned GraphSAGE node classification example with PyTorch Lightning (#3863)
* cleaned pl node classification example

* conform to PL's method of updating the dataloader

* update

* lint

* fix test

* fix
2022-04-12 17:26:23 +08:00
nv-dlasalle baa9292899 [bugfix] Fix EdgePredictionSampler for UVA sampling (#3904)
* Add failing unit test

* Fix negative sampler edge types

* fix test

* oops

* revert

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
2022-04-12 15:25:21 +08:00
Serge Panev f931c6ba54 [Examples] Add pure gpu mode in the GraphSAGE node classification and link prediction (#3856)
Signed-off-by: Serge Panev <spanev@nvidia.com>

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-12 13:03:03 +08:00
Xin Yao 5fcd7f290f [Feature] Enable UVA for GPU PinSAGE and RandomWalk (#3857)
* enable uva for pinsage sampler

* unit test

* modify some checks on the python side

* remove legacy random walk code

* update unit test

* update unit test

* fix unit test

* adjust checks

* move some checks to c++

* move max_nodes check to cuda kernel

* fix ci for tf

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-04-11 20:01:23 +08:00
Peiqi Yin 23a5e674e7 [fix] device cannot be Nonetype object. (#3822)
* fix device = none to cpu.

* fix singleton-comparison

* modify device is none

* fix use uva

* add docs

* add edge dataloader

* add doc

* fix trailing-whitespace

* modify default

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-11 14:58:27 +08:00
Mufei Li 43b9d40e5e [Doc] Fix Doc for EGNNConv (#3919)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-04-11 11:01:15 +08:00
Mufei Li 01e50626d3 [Example] Add TorchMetrics in README (#3913)
* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Rhett Ying <85214957+Rhett-Ying@users.noreply.github.com>
2022-04-09 23:25:51 +08:00
Rhett Ying adb3a7c11e [BugFix] record/restore pin status when pickle/unpickle (#3914)
* [BugFix] record/restore pin status when pickle/unpickle

* disable test on TF

* set version as expected

* unpin memory in test
2022-04-09 11:42:23 +08:00
Mufei Li ed15b47147 Update (#3916) 2022-04-08 14:29:03 +08:00
rudongyu dc5035b1f3 [NN] Add EGNN & PNA (#3901)
* add EGNN & PNA

* fix egnn issues

* fix pna issues

* update pna conv

* add doc strings

* update pnaconv

* fix unused args issue

* fix moment aggregation issue

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-04-07 17:44:57 +08:00
nv-dlasalle 27a6eb564c [Examples] Update graphsage multi-gpu example to use mutliple GPUs for validation and testing. (#3827)
* Update graphsage multi-gpu example to use mutliple GPUs for validation and
testing.

* Remove argmax

* Fix rebase error

* Add more documentation to example and simplify

* Switch to name shared memory

* Add comment about how training is distributed

* Restore iteration count

* fix munmap error reporting for better error messages

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-04-06 00:32:37 +08:00
Mufei Li 11e910b6f8 [Doc] Hotfix for GraphSAINT Doc (#3898)
* Update graphsaint.py

* Update graphsaint.py
2022-03-31 19:22:21 +08:00
Quan (Andy) Gan 35e66f426a [Bugfix] Fix UVA sampling with partially specified node types (#3897)
* fix uva with partial node types

* lint

* skip tensorflow unit test
2022-03-31 17:07:02 +08:00
nv-dlasalle 7a2f494373 [bugfix] Make source build work for systems where the default is python2 (#3718)
* Make compatible for system where the default is python2

* Update cmakelists with find package

* Switch to find_program

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-03-30 08:27:58 -07:00
Krzysztof Sadowski 4c7bd314b4 [Feature] Radius Graph (#3829)
* radius graph

* remove trailing whitespaces from docs

* disable invalid name for transform func

* disable radius graph module invalid name

* move pylint disable before init

* fix missing nodes from point set

* update docs indexing

* add compute mode as optional param

* radius graph test

* remove trailing whitespaces

* fix precision when comparing tensors

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-03-30 17:23:29 +08:00
rudongyu 290b7c25c2 [Transform] Random Walk Positional Encoding & Laplacian Positional Encoding (#3869)
* [Transform] Random Walk Positional Encoding & Laplacian Positional Encoding

* Update python/dgl/transforms/module.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>

* Update python/dgl/transforms/functional.py & python/dgl/transforms/module.py

* update python/dgl/transforms/functional.py

* update doc indexing

* update unit test

* support weighted graphs for random_walk_pe

* change reference

* fix linter errors

* fix datatype compatibility with MXNet & TF

* fix device errors

* fix result precision errors

* change tensorflow abs api

* reboost CI

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-03-30 15:59:28 +08:00
Mufei Li 9fee20b91d [Sampler] [Example] SAINTSampler and Simplify GraphSAINT Example (#3879)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-03-30 13:51:37 +08:00
Liang Wang 0dd500e386 [BugFix] Remove redundant reference of networkx package in pagerank.py (#3888)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-03-30 11:54:03 +08:00
Peiqi Yin d8375c10ec [BugFix] involve indices_devices for other dataset object. (#3810)
* involve indices_devices for other dataset object.

* modify raise error when device not found.

* remove empty line

* fix line too long
2022-03-30 11:50:03 +08:00
Cheng Wan fbbca994b7 [Feature] METIS Partition with Communication Volume Minimization (#3821)
* upd

* upd

* upd

* upd

* upd

* fix OpenMP compatibility issues

* typo

* partition

* misc

* fix typo

* num_parts=1

* import torch

* long

* print info

* print info

* print info

* upd

* remove debug code

* revert partition.py

* fix cut count

* fix cut count

* Revert "fix cut count"

This reverts commit 10926b4fd48f45c8f1ddb58be7db6c22e653effd.

* Revert "fix cut count"

This reverts commit 76465283bef093a2b4209ad70dd15d2437b2ec8a.

* type of deprecate

* typo in deprecate info

* fix typo

* use cv for partitioning

* CE

* no message

* revert

* typo

* add objtype

* no message

* fix bug

* fix bug

* fix bug

* ?

* semicolon

* drop tensors

* no message

* backward

* backward

* max op

* store X.shape

* th

* test

* Revert "test"

This reverts commit 92b3b2f64a3a1128590098fa03ce429c5466e6ce.

* test

* tolist

* debug

* to cuda

* tuple

* fix bug

* remove X

* no message

* fix bug

* workload balance

* Revert "workload balance"

This reverts commit d7f8e4a16ba2a7eabb4a9bb945523bfe6623e723.

* reverse

* Revert "reverse"

This reverts commit 8a71cf25685aa7d889b9b8881b46f7a16b7d6e6d.

* Revert "Revert "reverse""

This reverts commit 196b143932d5cf9813576ece7c990b63d322d063.

* Revert "Revert "Revert "reverse"""

This reverts commit cf9e89a07013582056e7cde235e51331aca7fa9c.

* no message

* Merge commit '5498cf051e9cf450dcb798ca09920bbde008a661'

# Conflicts:
#	python/dgl/distributed/partition.py

* Revert "Merge commit '5498cf051e9cf450dcb798ca09920bbde008a661'"

This reverts commit f79be2ad777897c7025b28308454cad81ad6bb27.

* fix bug

* third party

* no message

* try to avoid memory leak

* try to avoid memory leak

* avoid memory leak with no hope

* Revert "avoid memory leak with no hope"

This reverts commit c77befe9479f46758e744642f66dd209b50eef7d.

* no message

* Revert "no message"

This reverts commit 478cb28fe25fb1002b2f1dc202bb9bdaad8b2a56.

* del

* Revert "del"

This reverts commit 1b468e45ce646b400ff3ffa61a0b2da058b3bdfd.

* no message

* no message

* Revert "no message"

This reverts commit 92e4f5561ed42da0606618b2fff9f1ad5ed439d9.

* third party

* document

* Update metis_partition.cc

* Update metis_partition_hetero.cc

* Update metis_partition_hetero.cc

* Update partition.py

* Update partition.py

* Update partition.py

Co-authored-by: yzh119 <expye@outlook.com>
Co-authored-by: chwan-rice <54331508+chwan-rice@users.noreply.github.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2022-03-27 14:22:01 -07:00
Minjie Wang ae3316c846 [Doc] Update doc of dataloading package. (#3886)
* wip: dataloading doc

* update dataloading package doc and many others

* lint
2022-03-26 21:59:02 +08:00
Sai Kandregula 6752bd4515 [Example] layers number fix in GAT example (#3872)
* gat layers fix

* minor bug

Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-03-26 15:02:16 +08:00
Quan (Andy) Gan f758db386f [Bug] Fix dtype mismatch in heterogeneous DataLoader (#3878)
* fix

* unit test
2022-03-26 13:56:21 +08:00
István Ketykó e963256882 [Doc] Update message-api.rst (#3844)
fix #3812

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-03-25 23:22:20 +08:00
Tong He f73c7a648b [bugfix] Set stacklevel=2 for dgl_warning (#3816)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-03-25 23:21:05 +08:00
Quan (Andy) Gan 7d4160869c [Bug] Fix multiple issues in distributed multi-GPU GraphSAGE example (#3870)
* fix distributed multi-GPU example device

* try Join

* update version requirement in README

* use model.join

* fix docs

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-25 22:04:02 +08:00
Rhett Ying 16409ff852 [Doc] fix display issue of node/edge_split (#3858)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-25 09:18:08 +08:00
Quan (Andy) Gan a0e8cf0d3e [Bugfix] Fix UVA sampling with lazy feature slicing (#3862)
* fix uva sampling with features

* fix

* add is_listlike function to distinguish strings from sequence

* fix

Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-03-24 20:13:01 +08:00
Quan (Andy) Gan 223a3da5ae [Bugfix] Fix multiple bugs and code refactor (#3841)
* fix

* remove setcxx methods

* move pin flag to CSR and COO matrix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-24 18:24:49 +08:00
Sai Kandregula 12e97c54ac [Feature] get_attention parameter in GlobalAttentionPooling (#3837)
* get_attention parameter in GlobalAttentionPooling

* removed trailing whitespace

* lint fix

Co-authored-by: decoherencer <decoherencer@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-03-24 12:49:40 +08:00
Rhett Ying e9fd65e9c9 [BugFix] send rpc messages blockingly in case of congestion (#3867)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-24 09:51:45 +08:00
Xin Yao 172949d429 [Example] Fix multi-GPU RGCN example (#3871)
* fix multi-gpu rgcn example

* remove dgl.multiprocessing in turorials

* add a comment

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-23 09:28:36 -07:00
Mufei Li 61edb7980a [NN] Update GNNExplainer (#3848)
* Update

* Update

* Update
2022-03-23 18:40:49 +08:00
Mufei Li 8005978e1e [NN] Grouped reversible residual connections for GNNs (#3842)
* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update
2022-03-23 16:58:39 +08:00
Jinjing Zhou a3fd059537 Fix jenkins (#3875)
* try fix

* try fix

* try fix

* try fix

* Revert "try fix"

This reverts commit a3fa0b1e9c0ab892cc3a22acf3770903db8b14a7.

* try fix shared memory

* try fix shared memory

* try fix image version

* fix
2022-03-23 14:13:45 +08:00
Rhett Ying 480a4ae35c [BugFix] pass ntype/etype into partition book when node/edge_split (#3828)
* [BugFix] pass ntype/etype into partition book when node/edge_split

* fix test failure

* fix test failue on mxnet

* fix test failure
2022-03-14 15:11:01 -07:00
nv-dlasalle 57d2f31f20 [Examples] Add pure gpu example of graphsage (#3796)
* Add pure_gpu example of graphsage

* move to advanced directory

Co-authored-by: Quan Gan <coin2028@hotmail.com>
2022-03-10 17:33:23 -08:00
RecLusIve-F f908f35c38 [Example]Add P-GNN example (#3823)
* [Model]P-GNN

* updata

* [Example]P-GNN

* Update README.md

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2022-03-10 17:39:32 +08:00
paoxiaode eec219abbc Change the parameter of curand_init (#3794)
* Change the curand_init parameter

* Change the curand_init parameter

* commit

* commit

Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
2022-03-10 16:31:09 +08:00
Quan (Andy) Gan 520cef88df [Feature] Compatibility to DLPack 0.6 in tensoradapter (#3803)
* compatibility to DLPack 0.6 in tensoradapter

* fix

* oops

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2022-03-10 10:32:52 +08:00
Jinjing Zhou 27d3af01fb Fix benchmark problem (#3762)
* fix

* fix bench

* fix

* Update bench_rgcn.py

* try fix

* fix wrong fix

* Update bench_rgcn.py
2022-03-08 19:27:00 +08:00
Quan (Andy) Gan 44638b9336 fix ddp dataloader in heterogeneous cases (#3801) 2022-03-07 16:56:43 +08:00
Rhett Ying bb6cec23b4 [BugFix] fix incorrect name when fetch data in sparse optim (#3808) 2022-03-06 10:43:43 -08:00
Jinjing Zhou 8f0136b0e3 Remove unnecessary jenkins part (#3806)
* fix

* try mount

* what's happening

* fix

* ci
2022-03-04 17:21:19 +08:00