提交

提交图

3215 次代码提交

作者 SHA1 备注 提交日期
Quan (Andy) Gan b4cd60a952 fix relgraphconv bug (#3256) 2021-08-23 17:29:14 +08:00
Rhett Ying 8341244a2d [Doc] fix formula display issue in EdgeWeightNorm (#3275)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-21 21:59:26 +08:00
Jinjing Zhou f67b1908ec add flag (#3264)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-21 12:38:03 +08:00
Peiqi Yin f5b410b72d GeniePath model add a Tanh. (#3269)
* Update model.py

* Update README.md

Co-authored-by: Zihao Ye <expye@outlook.com>
2021-08-20 22:06:00 +08:00
nv-dlasalle 7f9279397a [Feature][DistDGL] Add NCCL support for range based partitions (#3213)
* Implement range based NDArrayPartition

* Finish implement range based partition support

* Add unit test

* Fix whitepace

* Add Kernel suffix

* Fix argument passing

* Add doxygen docs and improve variable naming

* Add unit test

* Add function for converting a partition book

* Add example to partition_op docs

* Fix dtype conversion for mxnet and tensorflow
2021-08-19 21:17:55 -07:00
Chen Kai 75fca8e4e8 [bugfix] fix the bug of DotGatConv when compute the edge_softmax (#3272)
Co-authored-by: Zihao Ye <expye@outlook.com>
2021-08-20 09:19:00 +08:00
nv-dlasalle f634950825 [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-19 12:24:15 -07:00
nv-dlasalle fc6f0b9efc [bugfix] Link against pytorch library by name rather than path (attempting to fix linking issue). (#3246)
* Use library name

* fix for mac builds from source

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-19 16:16:08 +08:00
maqy ea06688ed2 [Model] add model example EvolveGCN. (#3190)
* add evolveGCN example

* small fix

* fix defect

* fix defect

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-19 14:14:55 +08:00
Quan (Andy) Gan a536772ed2 fix cuda 11.1 crashing bug (#3265) 2021-08-18 21:16:49 +08:00
David Min 2613f7f013 [Performance] Cacheline-aligned access for UnifiedTensor (#3254)
* Add pytorch-direct version

* remove

* add documentation for UnifiedTensor

* Revert "add documentation for UnifiedTensor"

This reverts commit 63ba42644d4aba197c1cb4ea4b85fa1bc43b8849.

* alignment fix for UnifiedTensor access

* fix linting issue

Co-authored-by: shhssdm <shhssdm@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-18 01:29:13 +08:00
Zihao Ye 76af2a2ef3 [perf] Remove activation cache if not required. (#3258)
* upd

* fix

* upd
2021-08-17 17:14:51 +08:00
Eric Kim ac01e88082 [Tools] In tools/launch.py, correctly pass all DGL client/server env vars if udf is a multi-command (#3245)
* Correctly pass all DGL client/server env vars if udf is a multi-command

* Refactor to use wrap_cmd_with_local_envvars() helper fn
2021-08-16 18:54:48 -07:00
Jinjing Zhou 1d4d4f59a6 Fix regression test (#3257) 2021-08-16 15:30:30 +08:00
Peiqi Yin 68c0cfbb3c [Model] Fix diffpool loss (#3233) 2021-08-16 15:18:15 +08:00
blokhinnv 31772b142e fixed typos in docs and docstring (#3231)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-16 13:58:45 +08:00
xiang song(charlie.song) 6f36dd6366 Cleanup some code in graph_services.py (#3238)
* Fix bug

* Fix

* Fix

* upd

* Merge some code

* lint
2021-08-15 11:58:37 -07:00
Alex Morehead 1b350e9319 Fix small typo in knn_graph() documentation (#3244) 2021-08-12 14:23:27 +08:00
JOHNW02 738b75f41e [example] Create EEG-GCNN example. (#3186)
* Create EEG-GCNN example.

* Update README.md

* Remove gitignore file.

* Update README.md

* change 'datas' to 'datasets'.

* Change train.py to main.py

* Added an entry in the indexing page.

* State "simplified version"; change how to run.

* Fix bug in contact

* Remove paper link in reference.

* Create working branch

* Add normalization of x.

* Update paper link and tags

* Update paper link in readme

* Update readme; add patient level indices

* Update readme. Add comments to models

* Update README.md

* change to with; specify location for ch and el; move note

* fix bug for note

* Add args for models; clean code.

* delete = in readme

* Add reference for spec_coh_values
2021-08-11 13:47:59 +08:00
xiang song(charlie.song) cd2cf606db [Bugfix] Fix bug introduced by https://github.com/dmlc/dgl/pull/3131 (#3234)
* Fix bug

* Fix

* Fix

* upd

* trigger
2021-08-10 11:36:57 +08:00
nv-dlasalle 799c091e6d [bugfix] Only link the pytorch tensordispatcher against libtorch (#3225)
* Only link tensordispatcher against pytorch

* Only modify libraries when not using MSVC
2021-08-08 13:17:04 +08:00
xiang song(charlie.song) bcf92f6c21 [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-08-06 20:41:20 +08:00
nv-dlasalle 4d9beb3b7b [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-06 17:56:15 +08:00
Mufei Li 8c2f8ff84a Update citation_graph.py (#3221) 2021-08-06 15:57:42 +08:00
freeliuzc 45858d6803 [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-05 10:47:50 -07:00
Jinjing Zhou 8b5f4f5b1c Fix typos (#3214) 2021-08-03 17:39:45 +08:00
Yuchen 4ae13bd270 [bugfix] Fix force_reload parameter of FraudDataset (#3210)
* enable force_reload of FraudDataset

* rewrite hash_key in FraudDataset
2021-08-03 13:14:17 +08:00
Ankit Garg b131920066 [Feature] Add removed edges in distributed graph partitioning to handle heterogeneous graph (#3137)
* Added code for Rectifying (TypeError: unhashable type: 'slice') when copying file

* 1) added distributed preprocessing code to create ParMetis Input from CSV files
2) add code to run pm_dglpart on multiple machines
3) added support for recreating heteregenous graph from homo geneous graph based on dropped edges, as ParMetis currently only supports homogeneous graphs

* move to pandas

* Added comments and remove drop_duplicates as it was redundant

* Addressed Pr Comments

* Rename variable

* Added comment

* Added comment

* updated ReadMe

Co-authored-by: Ankit Garg <gaank@amazon.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2021-08-02 11:41:13 -07:00
David Min b78acd67ce [Doc] Add an official documentation of UnifiedTensor (#3194)
* Add pytorch-direct version

* remove

* add documentation for UnifiedTensor

* Revert "add documentation for UnifiedTensor"

This reverts commit 63ba42644d4aba197c1cb4ea4b85fa1bc43b8849.

* add UnifiedTensor documentation

* Update python/dgl/contrib/unified_tensor.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>

Co-authored-by: shhssdm <shhssdm@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-02 22:40:51 +08:00
nv-dlasalle f7ce267164 [bugfix] Fix curand_init() calls in rowwise sampling (#3196)
* Split out separate generators for each thread

* Amortize cost of curand_init

* Improve readability

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-02 19:08:11 +08:00
Eric Kim c40bbf4f33 [Tools] Refactor tools/launch.py to handle more python binary names (#3205)
* Refactors torch dist launcher udf-wrap code to handle more python versions

* minor changes
2021-08-02 17:31:02 +08:00
Quan (Andy) Gan 213e27f05c Bump nightly version (#3146) 2021-08-02 17:24:18 +08:00
Tianjun Xiao ddeb657f0d Add four dgl papers (#3212) 2021-08-02 17:10:36 +08:00
Quan (Andy) Gan 100eb57c45 [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-02 16:16:29 +08:00
xiang song(charlie.song) 0e90d04566 [Readme] Add pytorch direct to Awesome Papers Using DGL (#3211)
* Add pytorch direct to Awesome Papers Using DGL

* update
2021-08-02 15:24:52 +08:00
nv-dlasalle 6001001fed Improve usage of pinned memory in sparse_optimizer (#3207)
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2021-08-02 14:16:23 +08:00
David Min c793593577 [Feature] Add multi-GPU UnifiedTensor unit test (#3184)
* Add pytorch-direct version

* remove

* Add multi-gpu unified tensor test for pytorch

* relocate verification step to each process

* reduce number of workers

* add parameter

Co-authored-by: shhssdm <shhssdm@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-08-02 12:41:03 +08:00
Xiangkun Hu 50a62c19b6 [Doc] Add two papers to Awesome Paper List (#3208)
* 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

* Add awesome papers

* Fix index in paper list

* Add paper to paperlist

* add two papers

* add two papers

* Update README.md

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-08-02 11:57:01 +08:00
Jinjing Zhou 8bc91414e9 [Distributed] Fix distributed training hang with multiple samplers (#3169)
Rewrite the multiprocessing worker pool
2021-08-01 21:31:04 +08:00
Mufei Li 2ec0493df4 Update README.md (#3209) 2021-08-01 18:11:55 +08:00
Eric Kim efd0fc6cde [Tools] Adds --ssh_username to the dist launcher (tools/launch.py) (#3202) 2021-07-30 23:34:26 +08:00
Lu Lu 7ab659e257 Remove DGL_LOADALL=true (#3150)
DGL_LOADALL=true is not used in the code anymore.
2021-07-30 18:17:18 +08:00
Kay Liu 39764da491 [model] add model example GeniePath (#3199)
* [model] add model example GeniePath

* improvements based on feedback

* improvements based on feedback

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-07-30 18:09:42 +08:00
Jinjing Zhou a5dc230af9 Fix shared memory in unit test (#3204) 2021-07-30 15:02:09 +08:00
KounianhuaDu 6f93c6aa74 fix (#3167)
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-07-30 13:07:22 +08:00
Kay Liu a107993f10 [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-07-29 15:24:15 +08:00
Rhett Ying c4791fd46e [bugfix] fix default ntypes/etypes consistency between dgl.DGLGraph and dgl.graph (#3198) 2021-07-29 14:05:28 +08:00
Jinjing Zhou 2583ec5934 [Doc] Fix WeightBasis documentation (#3189)
* fix

* fix type

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-07-28 18:56:10 +08:00
Da Zheng cb4a83a233 [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-07-28 15:28:27 +08:00
Minjie Wang c3921599c8 Update README.md 2021-07-28 12:37:31 +08:00