mirrors / dmlc--dgl

提交

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

2937 次代码提交

此分支
此分支
所有分支
作者 SHA1 备注 提交日期
Mufei Li f0fafa2062 [Bug fix] Fix batch information with remove_nodes/edges applied (#3119) 
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix lint

* Fix

* Update

* Fix test cases

* Fix

* add docstrings

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-17.us-west-2.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
 2021-07-13 16:27:31 +08:00
Quan (Andy) Gan 7c3e1f94b9 Remove march=native flag (#3134) 2021-07-13 15:16:18 +08:00
Quan (Andy) Gan 2e19ba8025 add exclude self option for EdgeDataLoader (#3122) 
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-13 13:30:16 +08:00
Quan (Andy) Gan b576e617ad [Feature] Add left normalizer for GCN (#3114) 
* add left normalizer for gcn

* fix

* fixes and some bug stuff
 2021-07-13 12:49:35 +08:00
sanchit-misra fac75e166b [CPU][Kernel] Single socket spmm (#3024) 
* optimizations of spmm for CPU

* Added names of contributors

* Minor code cleanup

* Moved the spmm optimization code to a new header file

* Moved to DGL's logging method

* removed duplicate code between SpMMSumCsr and SpMMCmpCsr

* Changes made to follow Google coding style

* Fixed lint errors in spmm.h

* Fixed some lint errors from spmm_blocking_libxsmm.h

* Fixed lint errors from spmm_blocking_libxsmm.h

* Added comments to SpMMCreateLibxsmmKernel

* to enable building of tests, and other cosmetic changes

* disabling libxsmm on windows

* Put a condition to avoid opt impl for FP64 as libxsmm does not have FP64 support yet

* cosmetic changes and documentation

* cosmetic changes

* to pass lint tests

* replaced multiple allocations for buffers of indices and edges with a single allocation

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-13 11:18:29 +08:00
Rhett Ying 186ef59283 [Feature] apply dgl.reorder() onto several node classification datase… (#3102) 
* [Feature] apply dgl.reorder() onto several node classification datasets in DGL

* rebase on latest dgl.reorder_graph()
 2021-07-13 08:52:17 +08:00
Rhett-Ying 175f53decf [Feature] enable edge reorder in dgl.reorder_graph() (#3113) 
* [Feature] enable edge reorder in dgl.reorder_graph()

* refine doc string

* refine doc string for dgl.reorder_graph

* refine doc string further
 2021-07-12 14:11:46 +08:00
Kay Liu 0ce92a86ed [BugFix] fix problems found in bug bash (#3116) 
* fix breakline in fakenews.py

* fix inconsistent argument name

* modify incorrect example and deprecated graph type

* modify docstring and example in knn_graph

* fix incorrect node type

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-07-12 13:34:50 +08:00
Wilfried L. Bounsi c7449818a1 Update documentation for remove_edges function (#3123) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-07-12 12:54:21 +08:00
Jinjing Zhou 3641829211 fix docs (#3126) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-07-12 11:43:15 +08:00
Tomohiro Endo fc55225f08 Fix module name (#3128) 2021-07-12 11:15:02 +08:00
Quan (Andy) Gan 5798ee8d98 [Performance] Add a warning for ChebConv (#3099) 
* add a warning for chebconv

* fix and docstrings

* update bgnn

* fix

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
 2021-07-09 14:49:15 +08:00
Jinjing Zhou 183d29de56 fix cub problem (#3121) 2021-07-08 17:57:10 +08:00
Quan (Andy) Gan 2f41fcd986 [Build] fix various build problems (#3117) 2021-07-08 11:39:33 +08:00
Kay Liu bb89dee7ff [BugFix] fix problems in data split (#3082) 
* [BugFix] fix problems in data split

* fix format problems in docstring

* modify statistics to fit in dgl nature

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: zhjwy9343 <6593865@qq.com>
 2021-07-07 16:07:23 +08:00
Hengrui Zhang 0d1dcdcd2e [bugfix] fix a bug found in v0.7 bug bash in the 'grace' model (#3109) 
* Update aug.py

* Update aug.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2021-07-07 14:50:02 +08:00
Rhett-Ying 32d1f3ac97 [BugFix] skip frames whose num_rows is zero in dgl.heterograph.combine_frames() (#3110) 
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-07 14:15:06 +08:00
Quan (Andy) Gan 751685a979 [Doc] Add documentation for DGLGraph.adj_sparse (#3096) 
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-07 11:24:47 +08:00
Israt Nisa 188152b853 [Feature] Add Heterograph support on Python for builtin unary msg functions (copy_u, copy_e) (#2989) 
* heterograph for binary func

* Added SDDMM support

* Added unittest

* added binary test cases

* unary mfuncs works

* Fixed lint err

* lint check and others

* link check

* fixed import *_hetero issue

* lint check

* replace torch with dgl backend

* lint cehck

* removed torch from test

* skip mxnet unittest

* skip gpu test

* Remove unused/duplicated code

* minor

* changed data structure of ndata and edata

* link check

* reorganized

* minor lint

* minor lint

* raise error for udf func

* lint check

* fix for CUDA 10.1

* add a note for future cross-type max/min reducing

* Add support CUDA < 11

* lint check

* tidied C code

* remove dummy GSDDMM_hetero backward implementation

Co-authored-by: Israt Nisa <nisisrat@amazon.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan Gan <coin2028@hotmail.com>
 2021-07-06 20:41:58 +08:00
Jinjing Zhou 4556fab8e6 [Fix] Use logging to print warning message in socket creation (#3098) 
* fix

* fix

* lint

* fix

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-07-06 15:33:43 +08:00
Tianjun Xiao 22e8c120be [Example ] fix faiss issue on k 3 for hilander model (#3105) 
* fix faiss issue on k 3

* Add Hilander paper link

Co-authored-by: Tong He <hetong007@gmail.com>
 2021-07-06 13:22:56 +08:00
Zihao Ye c372f0aed6 [doc] Expose dgl.partition APIs (#3103) 2021-07-06 12:17:56 +08:00
Minjie Wang d3e4460b98 [Doc] Add docstring for missing APIs (#3088) 
* add docstring for missing API; fix some docstring

* rename apis; address comments
 2021-07-05 18:40:07 +08:00
Da Zheng 485c04cff6 [Distributed] Fix a few bugs in distributed API (#3094) 
* fix.

* fix.

* fix.

* fix.

* Fix test

Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-66.ec2.internal>
 2021-07-05 15:12:02 +08:00
Tianjun Xiao 595d4e3415 add SLAPS and Haystacks in awesome papers (#3097) 2021-07-05 14:16:10 +08:00
Quan (Andy) Gan ced0e443c2 [Doc] Add user guide for GPU-based sampling (#3070) 
* add user guide for gpu sampling

* Update minibatch-gpu-sampling.rst

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-05 12:07:43 +08:00
WangYQ 0851b4df5d Sageconv doc (#3093) 
* revise the doc of forward function in SAGEConv

* remove trailing space
 2021-07-05 09:25:56 +08:00
Tong He 9c41c22d7b [Model] Official implementation for HiLANDER model. (#3087) 
* add hilander model implementation draft

* use focal loss

* fix

* change data root

* add necessary scripts

* update download links

* update

* update example table

* fix

* update readme with numbers

* add empty folder

* only eval at the end

* set up hilander

* inform results may fluctuate

* address comments

Co-authored-by: sneakerkg <xiaotj1990327@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-212.us-east-2.compute.internal>
 2021-07-04 20:08:58 +08:00
唐宇謙 66ad774f9e [Tutorial] update typo in line 175 (#3092) 2021-07-03 18:14:30 +08:00
Da Zheng 0884d02465 [Distributed] Fix bugs in partitioning on heterogeneous graphs. (#3085) 
* fix bugs in partitioning on heterogeneous graphs.

* fix.

* fix.

* fix example.

* fix.

* fix test.

* fix.

* fix.

* fix.

* fix tests.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-71-112.ec2.internal>
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
 2021-07-02 21:16:14 +08:00
Rhett-Ying 04dce1e630 [Doc] refine doc and optimize metis part for dgl.reorder() (#3090) 
Co-authored-by: Tong He <hetong007@gmail.com>
 2021-07-02 20:14:47 +08:00
Jinjing Zhou 0b3a6216f5 [Test] Enable kvstore test (#3079) 
* try enable kvstore test

* fix

* fix

* seperate out kvstore test

* add comment
 2021-07-02 14:54:38 +08:00
ankit-garg 35ffceb409 Added code for Rectifying (TypeError: unhashable type: 'slice') when copying file (#3049) 
Co-authored-by: Ankit Garg <gaank@amazon.com>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
 2021-07-02 14:22:08 +08:00
Quan (Andy) Gan 4e74dc86af fix curand (#3077) 
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-07-02 13:37:06 +08:00
nv-dlasalle a0390dde93 [Feature] Add dgl.utils.is_sorted_srcdst() (#2685) 
* Add dgl.utils.is_sorted_srcdst

* Fix linting issues

* delete blank line

* Specify datatype to index tensor in test

* Force integer conversion

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-07-02 11:29:24 +08:00
Rhett-Ying 95f8ec83b5 [Doc] fix docstring in to_block() (#3081) 2021-07-02 09:45:45 +08:00
Tong He 1bfc3118e4 [Benchmark] Benchmark script for KNN Graph (#3080) 
* add knn benchmark result

* update

* update knn benchmark to test on multiple algorithms

* update benchmark script

* update
 2021-07-01 16:12:50 +08:00
Quan (Andy) Gan a97aaa4903 fix (#3076) 2021-06-30 18:27:05 +08:00
Quan (Andy) Gan 7823ae2dbb add TUDataset notice (#3075) 
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
 2021-06-30 14:22:45 +08:00
Rhett-Ying 4307fe8895 [Feature] Add dgl.reorder() to re-order graph according to specified … (#3063) 
* [Feature] Add dgl.reorder() to re-order graph according to specified strategy

* fix unit test failure for metis reorder

* fix unit test failure on mxnet_cpu

* refine unit test for dgl.reorder

* fix unit test failure on mxnet

* fix array_equal error for mxnet unit test

* fix unit test failure for mxnet

* convert metis output to numpy array explicitly

Co-authored-by: Tong He <hetong007@gmail.com>
 2021-06-30 13:29:08 +08:00
Quan (Andy) Gan 61fba9da0f [Doc] Add Node2Vec docstrings to documentation server (#3068) 
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
 2021-06-29 22:29:10 +08:00
Quan (Andy) Gan dad8751d29 [Doc] Add instructions for Windows users to install VC2015 redistributable (#3071) 
* add instructions for VC2015 redist install

* fix typo
 2021-06-29 13:10:48 +08:00
nv-dlasalle 02e5d47d2a [Example][Bugfix] Fix RGCN example datasplitting (#3037) 
* Fix example datasplitting

* Remove left-over split parameter

* Remove unused parameter

* Set epoch when using multiple GPUs

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2021-06-29 10:16:28 +08:00
Jinjing Zhou 2295c2186a [Benchmark] Add benchmark for GraphConv and HeteroGraphConv (#2999) 
* fix

* lint

* Revert "lint"

This reverts commit 263f913f02a9ece8491e9a9f812e50892da899ba.

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
 2021-06-28 17:16:18 +08:00
Tong He 789b03f433 [Doc] Add KNN Benchmark Results (#3055) 
* add knn benchmark result

* update
 2021-06-28 16:13:56 +08:00
Kay Liu 81456af8c5 [Doc] fix inconsistent variable names (#3065) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-06-28 13:30:08 +08:00
Tomohiro Endo 7f4086da7a cuda() to to(device) (#3064) 
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
 2021-06-28 11:56:24 +08:00
WangYQ 2f43cdb3c8 deal with situation where num_layers equals 1 (#3066) 2021-06-28 10:17:16 +08:00
Jinjing Zhou 9664cdffd3 [Build] Make nccl optional (#3056) 
* fix

* remove nvidiasmi

* fix

* fix docs

* fix

* fix

* 1

* fix

* remove

* skip deprecated kernel

* fix

* Revert "skip deprecated kernel"

This reverts commit c5ceb7f60dbbaf065b81cc3680757fd611d90ad3.

* fix
 2021-06-27 22:36:01 +08:00
Jinjing Zhou 115ac0b9a3 [Benchmark] Enable sample neighbors gpu benchmark (#3060) 
* skip deprecated kernels

* Revert "skip deprecated kernels"

This reverts commit 2ee7f212d11d0858a0f863349a5e02ad04e4bc3c.

* fix
 2021-06-25 14:24:01 +08:00