文件历史

982 次代码提交

作者 SHA1 备注 提交日期
Quan Gan 0d1807ecf4 update version 2021-11-05 03:12:27 +00:00
Quan (Andy) Gan ef1b7175d5 fix compatibility with PyTorch 1.10 (#3454) 2021-11-04 08:20:44 +00:00
Jinjing Zhou 87a4cb363a Fix #3437 (#3440) 2021-11-04 08:20:22 +00:00
Cheng Wan c7f94dbf5a [BugFix] Avoid Memory Leak Issue in PyTorch Backend (#3386)
* 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

* Update sparse.py

* Update tensor.py

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-11-04 08:18:40 +00:00
nv-dlasalle a63ad14bda [PyTorch][Bugfix] Use uint8 instead of bool in pytorch to be compatible with nightly version (#3406)
* Use uint8 instead of bool in pytorch

* Handle type aliases

* Fix syntax error

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-11-04 08:15:46 +00:00
Quan (Andy) Gan bf11b19e76 Update dataloader.py (#3322) 2021-11-04 08:14:21 +00:00
Rhett Ying 23d575ba81 [Bug] Do not skip graphconv even no edge exists (#3416) 2021-11-04 06:14:10 +00:00
Rhett Ying ee183f69a1 [BugFix] add count_nonzero() into SA_Client (#3417) 2021-11-04 06:13:40 +00:00
Rhett Ying cc83b49e01 [Bug] check dtype before convert to gk (#3414) 2021-11-04 06:13:19 +00:00
Rhett Ying 56b04b2125 [BugFix] extract gz into target dir (#3389) 2021-11-04 06:11:41 +00:00
Kay Liu 6df6a0881e [BugFix] fix typo in fakenews dataset variable name (#3363) 2021-11-04 06:09:43 +00:00
jwyyy 9b597b8e74 fix broadcast tensor dim (#3351)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-11-04 06:09:09 +00:00
xiang song(charlie.song) f9d51fdf0b [Feature] Add a HINT for the per edge type sampler of heterogeneous DistGraph that highlighting the etypes are sorted already. (#3260)
* pass cpp test

* distgraph use sorted edge flag.

* lint

* triger

* update test

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
2021-11-04 06:02:42 +00:00
xiang song(charlie.song) 269f90428c [Bugfix] Fix #3291 (#3333)
* Fix #3291

* update

* fix

* Unit key

* Fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-11-04 05:49:24 +00:00
esang f77bee328a [Bugfix] Fix bugs of farthest_point_sampler (#3327)
* fix start_idx

* fix the bug when cuda > 0

Co-authored-by: Tong He <hetong007@gmail.com>
2021-11-04 05:48:21 +00:00
Da Zheng 95b1eb734f [Distributed] Fix a bug in sampling an empty frontier (#3298)
* handle empty frontiers.

* fix lint.

* fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-11-02 09:51:03 +00:00
xiang song(charlie.song) 6b6174a5c4 Cleanup some code in graph_services.py (#3238)
* Fix bug

* Fix

* Fix

* upd

* Merge some code

* lint
2021-11-02 09:50:37 +00:00
xiang song(charlie.song) 4010e20cd9 [Bugfix] Distributed training can not work with dgl.dataloading.negative_sampler (#3215)
* Fix dist negative data loader bug

* upd

* Fix

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-11-02 09:23:21 +00:00
xiang song(charlie.song) 0246bf5567 [Bugfix] Fix bug introduced by https://github.com/dmlc/dgl/pull/3131 (#3234)
* Fix bug

* Fix

* Fix

* upd

* trigger
2021-11-02 09:22:52 +00:00
Da Zheng f0903275e8 [Distributed] Enable distributed EdgeDataLoader (#3192)
* make heterogeneous find_edges

* add distributed EdgeDataLoader.

* fix.

* fix a bug.

* fix bugs.

* add tests on distributed heterogeneous graph sampling.

* fix.

Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
2021-11-02 09:22:10 +00:00
xiang song(charlie.song) 508197e807 [New Feature] Per edge type sampler for to_homogeneous graphs. (#3131)
* fix.

* fix.

* fix.

* fix.

* Fix test

* Deprecate old DistEmbedding impl, use synchronized embedding impl

* Basic imple of heterogeneous on homogenenous sampling

* make pass

* Pass C++ test

* Add python test code

* lint

* lint

* Add MultiLayerEtypeNeighborSampler

* Add unitest for single machine dataloader

* Add dist dataloader test for edge type sampler

* Fix lint

* fix

* support for per etype sample

* Fix some bug and enable distributed training with per edge sample

* fix

* Now distributed training works

* turn off some mxnet

* turn off mxnet for some dist test

* fix

* upd

* upd according to the comments

* Fix

* Fix test and now distributed works.

* upd

* upd

* Fix

* Fix bug

* remove dead code.

* upd

* Fix

* upd

* Fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-11-02 09:21:20 +00:00
Da Zheng 223795eb82 [Distributed] Distributed heterograph training (#3069)
* support hetero RGCN.

* fix.

* simplify code.

* sample_neighbors return heterograph directly.

* avoid using to_heterogeneous.

* compute canonical etypes in advance.

* fix tests.

* fix.

* fix distributed data loader for heterograph.

* use NodeDataLoader.

* fix bugs in partitioning on heterogeneous graphs.

* fix lint.

* fix tests.

* fix.

* fix.

* fix bugs.

* fix tests.

* fix.

* enable coo for distributed.

* fix.

* fix.

* fix.

* fix.

* fix.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
2021-11-02 09:18:15 +00:00
Quan Gan 2b80161865 update version 2021-08-26 17:33:50 +00:00
Konstantinos Vandikas eca9b8ef08 allow for configuring default_dir (#3277)
* allow for configuring default_dir

* allow for using DGLDEFAULTDIR environment variable

* Update env_var.rst

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-26 17:32:16 +00:00
Rhett Ying d9e2618b20 [Doc] fix formula in RelGraphConv (#3285)
* [Doc] fix formula in RelGraphConv

* refine descriptions
2021-08-26 17:32:10 +00:00
Quan (Andy) Gan 1fa3202bbd fix relgraphconv bug (#3256) 2021-08-26 17:31:41 +00:00
Rhett Ying 50b87596b8 [Doc] fix formula display issue in EdgeWeightNorm (#3275)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-26 17:31:35 +00:00
Chen Kai b7af145d0b [bugfix] fix the bug of DotGatConv when compute the edge_softmax (#3272)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-08-20 02:47:54 +00:00
nv-dlasalle d9a2d74167 [Performance][Feature] Implement edge excluding in EdgeDataLoader on GPU (#3226)
* Update filter code

* Add unit tests

* Fixes

* Switch to indices

* Rename functions

* Fix linting

* Fix whitespace

* Add doc

* Fix heterograph

* Change workspace allocation

* Fix linting

* Fix docs in filter.py

* Add todo

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-20 02:47:48 +00:00
Zihao Ye 92550b3b03 [perf] Remove activation cache if not required. (#3258)
* upd

* fix

* upd
2021-08-20 02:44:06 +00:00
blokhinnv 731660d23e fixed typos in docs and docstring (#3231)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-20 02:41:21 +00:00
Alex Morehead 43c6ccc9bd Fix small typo in knn_graph() documentation (#3244) 2021-08-20 02:40:05 +00:00
nv-dlasalle 55dfef0e81 [Fix] FIx check for num_workers for using ScalarDataBatcher (#3219)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-20 02:36:53 +00:00
Mufei Li 72b95fe186 Update citation_graph.py (#3221) 2021-08-20 02:36:20 +00:00
freeliuzc 7466f9b1b8 [feature] add count_nonzero function for DistTensor (#3203)
* add count_nonzero function for DistTensor

* change the load method of local data

Co-authored-by: liuzichang04 <liuzichang04@meituan.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-20 02:36:15 +00:00
Yuchen b5a1eea5ed [bugfix] Fix force_reload parameter of FraudDataset (#3210)
* enable force_reload of FraudDataset

* rewrite hash_key in FraudDataset
2021-08-20 02:33:19 +00:00
Quan (Andy) Gan 526d4ac0bb [Doc] Fix random_walk documentation inconsistency (#3188)
* Update randomwalks.py

* Update randomwalks.py

FIx lint

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-20 02:26:20 +00:00
nv-dlasalle 988cf98e43 Improve usage of pinned memory in sparse_optimizer (#3207)
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-20 02:25:43 +00:00
Jinjing Zhou dfe422dd33 [Distributed] Fix distributed training hang with multiple samplers (#3169)
Rewrite the multiprocessing worker pool
2021-08-20 02:24:06 +00:00
Kay Liu fb5464df76 [Model] add model example CARE-GNN (#3187)
* [Model] add model example CARE-GNN

* update README

* improvements based on the review feedback

* fix missing item()

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-08-20 02:19:12 +00:00
Rhett Ying 392a2e587d [bugfix] fix default ntypes/etypes consistency between dgl.DGLGraph and dgl.graph (#3198) 2021-08-20 02:18:59 +00:00
Lu Lu 35c898c9e9 Bug fix: change th to tf (#3182)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-20 02:14:19 +00:00
Xiangkun Hu 5f5a6ef4f7 [Dataset] Fix the edge numbers in the GNNBenchmarkDataset doc string (#3163)
* 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 doc string

* fix readme

* fix readme

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-07-21 11:07:51 +00:00
Quan Gan 86403757b7 update version 2021-07-19 01:58:21 +00:00
Quan Gan d539bb3d53 update version 2021-07-16 06:29:27 +00:00
Quan Gan 18ddbb0ad3 update version 2021-07-16 06:29:27 +00:00
Jingcheng Yu b379dbd6c8 Optimize dist_graph/_split_even_to_part memory usage (#3132)
Co-authored-by: yujingcheng02 <yujingcheng02@meituan.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-07-15 11:51:56 +08:00
Rhett Ying 5cdf7cfa98 [Doc] refine metis part of dgl.reorder_graph() (#3136)
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-07-15 10:14:39 +08:00
Mufei Li 3f6f694159 [Bug fix] Various fix from bug bash (#3133)
* Update

* Update

* Update dependencies

* Update

* Update

* Fix ogbn-products gat

* Update

* Update

* Reformat

* Fix typo in node2vec_random_walk

* Specify file encoding

* Working for 6.7

* Update

* Fix subgraph

* Fix doc for sample_neighbors_biased

* Fix hyperlink

* Add example for udf cross reducer

* Fix

* Add example for slice_batch

* Replace dgl.bipartite

* Fix GATConv

* Fix math rendering

* Fix doc

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-17.us-west-2.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-22-156.us-west-2.compute.internal>
2021-07-15 09:17:15 +08:00
Quan (Andy) Gan 066bef21f0 [Doc] Fix CoraFull number of edges inconsistency in doc (#3139) 2021-07-14 15:06:57 +08:00