mirrors / dmlc--dgl

提交

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

2937 次代码提交

此分支
此分支
所有分支
作者 SHA1 备注 提交日期
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
Rhett Ying 0528e90d7d [Examples] fix path for load_graph (#3797) 2022-03-03 15:36:48 +08:00
Rhett Ying 1d80d91a32 [BugFix] call g.is_homogenous instead of len(g.etypes) (#3793) 2022-03-02 14:27:52 +08:00
Quan (Andy) Gan 7f1a5e080b bump version (#3795) 2022-03-02 13:02:55 +08:00
Quan (Andy) Gan 396d71809c [Build] Working around broken name mangling in MSVC 16.5.5 + CUDA 11.3 (#3790) 
* fix

* explain

* oops
 2022-03-01 22:59:23 +08:00
Rhett Ying 0ec439247a [Examples] re-locate load_graph for share (#3784) 2022-03-01 09:32:53 +08:00
Da Zheng e0f0fa2a78 Fix a bug in sparse optimizer. (#3779) 
* fix a bug.

* Update pylintrc

* fix.

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2022-02-28 12:04:32 -08:00
Quan (Andy) Gan 3521fbe9ae [Build] Split spmm.cu and sddmm.cu for building on Windows (#3789) 
* split files

* fix
 2022-03-01 03:04:51 +08:00
Quan (Andy) Gan 6e1c69904e [Build or bug?] Fix VS2019 compilation error in randomwalk GPU kernel (#3788) 
* Update randomwalk_gpu.cu

* Update randomwalk_gpu.cu
 2022-02-28 22:02:23 +08:00
Xin Yao 8425c936c2 update doc for gpu&uva sampling (#3787) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2022-02-28 21:17:32 +08:00
Xin Yao 861666fa55 [Bugfix] Fix RGCN Example (#3785) 2022-02-28 18:55:06 +08:00
Mufei Li ec64581c96 [Doc] Misc fix for doc (#3782) 
* Update nn-pytorch.rst

* Update dgl.data.rst

* Update dgl.rst
 2022-02-28 16:03:32 +08:00
Quan (Andy) Gan 62e23bd5b1 [Bugs] Fix distributed example error and import error (#3783) 
* fix

* raise an error

* fix docserver crash
 2022-02-28 14:58:50 +08:00
Jinjing Zhou 7ab10348ed Sort recipe in DGL-Go (#3781) 
* add

* remove

* fix

* rework the readme and some changes

* add png

* update png

* add recipe get

* sort recipe list and remove copy

* fix

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2022-02-28 13:27:02 +08:00
Minjie Wang 869991d7dc [Doc] Update README; add DGL-Go related 2022-02-28 11:44:55 +08:00
Minjie Wang e36cada2e8 [Doc] Update readme for DGL-Go 2022-02-28 10:53:16 +08:00
Jinjing Zhou 266b21e535 [DGL-Go] Change name to dglgo (#3778) 
* add

* remove

* fix

* rework the readme and some changes

* add png

* update png

* add recipe get

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2022-02-28 10:40:01 +08:00
Quan (Andy) Gan d41d07d0f6 [Doc and bugfix] Add docs and user guide and update tutorial for sampling pipeline (#3774) 
* huuuuge update

* remove

* lint

* lint

* fix

* what happened to nccl

* update multi-gpu unsupervised graphsage example

* replace most of the dgl.mp.process with torch.mp.spawn

* update if condition for use_uva case

* update user guide

* address comments

* incorporating suggestions from @jermainewang

* oops

* fix tutorial to pass CI

* oops

* fix again

Co-authored-by: Xin Yao <xiny@nvidia.com>
 2022-02-28 02:21:51 +08:00
Mufei Li 3bd5a9b6d1 [Doc] Fix Doc (#3777) 
* Update

* Update

* Update

* Update
 2022-02-25 17:21:06 +08:00
Mufei Li 6d9433b04f [Transform] [Doc] Rename transform to transforms and update doc (#3765) 
* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix
 2022-02-24 21:28:15 +08:00
Rhett Ying ccaa0bf282 [Doc] refine docstring and examples for reorder_graph (#3770) 
* [Doc] refine docstring and examples for reorder_graph

* refine docstring
 2022-02-24 17:31:00 +08:00
Rhett Ying 609a49a9ba [Doc] fix example in dataset adapter docstring (#3772) 2022-02-24 16:57:25 +08:00
Rhett Ying 2f4146a4c2 [Dataset] add train_idx/val_idx/test_idx as dataset attributes (#3769) 
* add train_idx/val_idx/test_idx as dataset attributes

* refine docstring
 2022-02-24 15:17:40 +08:00
sanchit-misra e7ad4c9c8b Fixes the bug when total_nnz is > integer limit (#3766) 2022-02-23 13:34:34 -08:00
Rhett Ying 4b182b9063 toctree typo for dataset adapters (#3764) 
* toctree typo for dataset adapters

* Update docs/source/api/python/dgl.data.rst
 2022-02-23 18:46:19 +08:00
Rhett Ying 8db2dd3355 [Dataset][Fix] Allow non-numeric values and some fix on doc (#3757) 
* [Fix] be able to parse ids if numeric and non-numeric values are used together

* add required package info and cache note into docstring

* duplicate node id is not allowed
 2022-02-23 17:26:12 +08:00