提交

792 次代码提交

作者 SHA1 备注 提交日期
Quan Gan 55e056fbae update to 0.4.2 0.4.2 2020-01-20 17:35:24 +08:00
Quan Gan 10d2b8408d Merge branch 'master' into 0.4.x 2020-01-20 17:33:42 +08:00
Tong He f1a8f926a4 [Feature] Add environment variable to switch on/off MXNet set_np_shape (#1207)
* add env var

* Trigger CI

* simplification

* add doc
2020-01-20 13:54:14 +08:00
VoVAllen a00636a02b [Tensorflow] Several nn & example (#1191)
* several nn example

* appnp

* fix lint

* lint

* add dgi

* fix

* fix

* fix

* fff

* docs

* 111

* fix

* change init

* change result

* tiaocan+1

* fix

* fix lint

* fix

* fix
2020-01-19 22:48:40 +08:00
xiang song(charlie.song) 31a7d50964 [Bug Fix] Add boundary check for heterograph build with card as input (#1192)
* Add boundary check for heterograph build with card as input

* Fix when u or v is empty

* fix test_kernel.py error print

* Revert "fix test_kernel.py error print"

This reverts commit a71c20292549c0ac62a5326c30669ca4bde8febc.

* Turn op validation check on graph and bipartite by default

* upd

* udp

* upd

* update test
2020-01-18 17:18:40 +08:00
xiang song(charlie.song) c49582c996 [Test] Fix the random seed in test_kernel.py and fix the _print_error (#1194)
* Fix the random seed and fix the _print_error

* Fix check_positive_edge_sampler sample weight.

Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-17 13:48:47 +08:00
Quan (Andy) Gan 64f6f3c1a8 [Bug?] Fix non-determinism when setting the same seed across different runs with single thread (#1205)
* fix multithread random seed value

* more fix

* typo

* what did i wrote
2020-01-16 15:03:26 +08:00
Quan (Andy) Gan 86fe58ebb8 [Bug] Another fix on PyTorch memory leakage issue (#1139)
* another fix

* another try

* fix

* rewriting with kernel functions

* revert mxnet softmax changes

* lint fix

Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
2020-01-14 22:54:19 +08:00
Mufei Li a199b58a6a Fix (#1195) 2020-01-12 02:47:29 +08:00
xiang song(charlie.song) cd9599ab04 [KG][Optimization] Remove copy from parent in minibatch generation (#1193)
* remote copy from parent

* upd

* make test_sampler easier to pass
2020-01-10 15:59:09 -08:00
Quan (Andy) Gan 28a24414c8 [BUG] Fix empty heterograph creation with list & tuple (#1188)
* [BUG] Fix empty heterograph creation with list & tuple

* more fix
2020-01-09 22:40:27 +08:00
Zihao Ye aa06e4579f [Doc] Improve the docstring of several nn modules. (#1187)
* upd

* upd

* upd

* upd

* lint

* upd

* upd

* upd

* upd

* upd

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-09 21:14:33 +08:00
Quan (Andy) Gan 3ef757dbff [Feature] Pickle support for heterograph (#1177)
* [Feature] Pickle support for heterograph

* fix lint

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-09 13:27:29 +08:00
Zihao Ye ca302a13ac [Feature] Add dgl.nn.*.Sequential for usability (#1166)
* upd

* upd

* upd

* upd

* lint

* upd

* upd

* upd

* upd

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-09 12:39:22 +08:00
Quan (Andy) Gan 3d6646288f Update README.md (#1184) 2020-01-09 12:06:48 +08:00
Quan (Andy) Gan bfc2d3030d [BUG] Fix MovieLens feature assignment (#1183)
Fixes #1182 .

This is embarrassing.
2020-01-09 11:21:18 +08:00
xiang song(charlie.song) bb6a64763e [Feature][KG] Multi-GPU training support for DGL KGE (#1178)
* multi-gpu

* Pytorch can run but test has acc problem

* pytorch train/eval can run in multi-gpu

* Fix eval

* Fix

* Fix mxnet

* trigger

* triger

* Fix mxnet score_func

* Fix

* check

* FIx default arg

* Fix train_mxnet mix_cpu_gpu

* Make relation mix_cpu_gpu

* delete some dead code

* some opt for update

* Fix cpu grad update
2020-01-08 18:07:04 +08:00
Zihao Ye f1420d194f [Refactor] Renaming class methods of sampler utilities to improve readability (#1180)
* upd

* upd
2020-01-07 22:02:44 +08:00
VoVAllen 319118341d [Dataset] Change download repo link to https://data.dgl.ai (CDN) (#1175)
* change download link to cdn

* Update env_var.rst
2020-01-07 17:16:45 +08:00
Chao Ma 38d292e541 [KVStore] Fix memory leak bug. (#1174)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* update

* fix lint

* delete msg

* clear kv msg

* update

* update

* update

* update

* update

* update

* is not None

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2020-01-06 19:47:54 +08:00
Quan (Andy) Gan e705118bcc some comments (#1132) 2020-01-06 10:36:11 +08:00
Chao Ma 6d0a9c4297 [Fix issue] Fix issue 1170 (#1171)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* update

* fix lint
2020-01-05 14:31:41 +08:00
Da Zheng 1022d5d64d [KG] PBG's way of constructing negative edges (#1159)
* attach positive.

* add neg_deg_sample.

* add comment.

* add neg_deg_sample for eval.

* change the edge sampler.

* rename edge sampler in KG.

* allow specifying chunk size and negative sample size separately.

* fix bugs in KG.

* add check in sampler.

* add more checks.

* fix

* add comment.

* add comments.
2020-01-04 19:25:06 -08:00
Da Zheng 1de192f445 [KG] More fixes on eval (#1168)
* remove parallel sampling for multiprocessing.

* avoid memory copy in eval.

* remove print.
2020-01-04 13:33:13 -08:00
Da Zheng 346bc23556 [KG] Disable filter in evaluation (#1162)
* add no_eval_filter

* fix eval.
2020-01-03 13:02:34 +08:00
Da Zheng 7451bb2a01 merge eval results in all processes. (#1160) 2020-01-01 21:40:31 -08:00
Zihao Ye dfb10db86f [Feature] Bag of samplers for efficient categorical sampling w/ and w/o replacement (#1142)
* upd

* upd

* lint

* upd

* upd

* lint

* upd

* refactor

* lint

* upd

* upd

* upd

* lint

* fix

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* complete test

* upd

* upd

* fix

* vir

* upd

* fix

* fuck

* fix numerical
2020-01-01 11:42:11 +08:00
Da Zheng 162dc19a5d Accelerate edge sampling (#1151)
* accel edge sampling.

* measure sampling time.

* attach positive.

* add neg_deg_sample.

* fix a minor bug.

* remove unnecessary code.

* Revert "add neg_deg_sample."

This reverts commit 45f3c08c87e33589563ca7e6a3ec91cb1239910e.

* Revert "attach positive."

This reverts commit 503d7821555b69b7ccfed55ce8b87ee24fff852d.

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2019-12-31 01:40:20 -08:00
xiang song(charlie.song) 61b78e6ee7 [Bug Fix]edge sample hotfix (#1152)
* hot fix

* Fix docs

* Fix ArrayHeap float overflow bug

* Fix

* Clean some dead code

* Fix

* FIx

* Add some comments

* run test
2019-12-30 19:42:57 -08:00
Chao Ma 6731ea3af7 [KVStore] Reduce memory cost of kvstore (#1156)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint
2019-12-30 20:25:32 +08:00
Chao Ma 4f02bb751b [KVStore] Made kvstore can support multiple NICs (#1150)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update
2019-12-30 15:17:31 +08:00
Da Zheng f81841532f [KG] reduce memory consumption. (#902)
* reduce memory consumption.

* fix a bug.

* fix a bug.

* fix.
2019-12-28 21:55:50 -08:00
Quan (Andy) Gan 655d75688a fix deprecation warning (#1147) 2019-12-28 12:42:16 +08:00
xiang song(charlie.song) 913e324903 Add device check for sampler input (#1145)
current samplers only support working on CPU
2019-12-27 22:50:52 +08:00
Chao Ma d57ff78da1 [Small Fix] Change default message queue size of communicator to 2GB (#1140)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1
2019-12-25 17:03:52 +08:00
Minjie Wang 17aab8128a [Doc] update working with multiple backend section (#1128)
* update work with different backend section

* fix some warnings

* Update backend.rst

* Update index.rst

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2019-12-24 17:32:56 +08:00
Chao Ma e4ef8d1a80 [KVStore] API change of kvstore (#1058)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument
2019-12-24 17:05:15 +08:00
xiang song(charlie.song) 41f8a162f4 [Feature Improvement]Edge Sampler w and w/o replacement (#1116)
* Add weight based edge sampler

* Can run, edge weight work.
TODO: test node weight

* Fix node weight sample

* Fix y

* Update doc

* Fix syntex

* Fix

* Fix GPU test for sampler

* Fix test

* Fix

* Refactor EdgeSampler to act as class object not function that it
can record its own private states.

* clean

* Fix

* Fix

* Fix run bug on kg app

* update

* update test

* test

* Simply python API and fix some C code

* Fix

* Fix

* Fix syntex

* Fix

* Update API description

* add replacement for edge sampler

* Now edge sampler support replacement and no-replacement

* Fix

* Fix

* change kg/app to use edge sampler with replacement config

* Update replacement algo

* Fix syntax

* Update

* Update

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2019-12-24 16:25:40 +08:00
Da Zheng e890a899fc Partition a graph with HALO nodes (#1076)
* get subgraph with halo.

* add partition function.

* add comment.

* parallel partition.

* fix a compilation error.

* fix lint error.

* address comments.

* add comments.

* fix for TF.
2019-12-23 08:46:53 -08:00
VoVAllen 0d9acc9d58 hot fix (#1135) 2019-12-24 00:19:25 +08:00
VoVAllen a1b28b3954 [Hot fix] Fix build threads (#1133)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image

* try node

* try

* fix

* try new ci

* try new ci

* try new ci

* try new ci

* ci

* try

* try

* fix

* hot fix

* fix

* fix cpp test

* add comments
2019-12-23 21:42:59 +08:00
Mufei Li 2ab4d0e10f [Doc] Hotfix for README (#1119) 2019-12-23 19:22:38 +08:00
VoVAllen 7b98e899f3 [CI] Try new CI (#1124)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image

* try node

* try

* fix

* try new ci

* try new ci

* try new ci

* try new ci

* ci

* try

* try

* fix
2019-12-23 17:28:06 +08:00
VoVAllen d30a69bf27 [Backend] TF backend (#978)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image
2019-12-20 15:56:51 +08:00
Mufei Li cf9ba90fd4 [Chem] ACNN and various utilities (#1117)
* Add several splitting methods

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Finally

* CI
2019-12-19 11:54:40 +08:00
Zihao Ye 6beef85b07 upd (#1113) 2019-12-17 15:24:04 +08:00
John Andrilla 41114b8758 [Doc] Tree-LSTM in DGL, edit pass for readability (#1031)
* [Doc] Tree-LSTM in DGL, edit pass for readability

Edit for grammar and style. 
How about deleting the last line? "Besides, you..." It needs a transition or some context.

* Update tutorials/models/2_small_graph/3_tree-lstm.py

* Update tutorials/models/2_small_graph/3_tree-lstm.py
2019-12-17 14:14:18 +08:00
xiang song(charlie.song) 632a9af8f9 [Feature] No-Uniform Edge Sampler (#1087)
* Add weight based edge sampler

* Can run, edge weight work.
TODO: test node weight

* Fix node weight sample

* Fix y

* Update doc

* Fix syntex

* Fix

* Fix GPU test for sampler

* Fix test

* Fix

* Refactor EdgeSampler to act as class object not function that it
can record its own private states.

* clean

* Fix

* Fix

* Fix run bug on kg app

* update

* update test

* test

* Simply python API and fix some C code

* Fix

* Fix

* Fix syntex

* Fix

* Update API description
2019-12-16 11:26:34 -08:00
VoVAllen dd65ee211e [CI] Change tests for flexibility
* change ci image

* fix

* force bash

* fix

* fix python version

* fix

* fix

* fix

* update gpu

* cuda

* jenkins

* fix build sh

* fix

* Revert "fix"

This reverts commit 6b091914b3ef6b4300fa662ee375aa4b27944f5c.

* try fix

* fix

* Revert "fix"

This reverts commit e42c3035fa4974c6b197aaba0748f7b45113ddaa.

* try fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix tests

* try fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix ctx problem

* fix many tests

* fix typo

* add backend

* move to pytorch folder

* fix?

* fix ci

* try skip

* try false

* try?

* try?

* Revert to 7d9a52f

* fix various

* fix lint

* Revert Jenkinsfile

* revert env

* revert env

* address comment

* remove file
2019-12-16 14:54:37 +08:00
John Andrilla f36a451420 [Doc] Old wines new title, edit for grammar and style (#1027)
* Old wines new title, edit for grammar and style

new descriptive title here and an edit pass @aaronmarkham

* Update tutorials/models/4_old_wines/README.txt
2019-12-16 11:24:58 +08:00