文件历史

2100 次代码提交

作者 SHA1 备注 提交日期
Quan (Andy) Gan 3e72c53ae0 [Bug?] Fix #1563 (#1642) 2020-06-15 22:00:21 +08:00
Quan (Andy) Gan ba2ee7bdcd [Bug] fixing from_edge_list (#1615)
Co-authored-by: Chao Ma <mctt90@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-06-15 18:12:55 +08:00
Minjie Wang d6d517bb02 [Kernel] CUDA CSR2COO COOSort COO2CSR (#1620)
* add cuda source

* moving codes from kernel2 branch

* operator overloading

* Better error message for unsupported device

* fix c tests

* coo sort using cusparse

* move test_rpc to distributed

* lint

* address comments and add utests

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Chao Ma <mctt90@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-06-15 14:35:12 +08:00
xiang song(charlie.song) 61f007c43b [NN] Add low memory support for nn.RelGraphConv (#1631)
* Add low memory support for relgraphconv
y

* lint

* Add tf support

* Fix lint

* Fix mxnet

* Fix mxnet

* Fix

* minor fix

* Fix

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
2020-06-15 13:49:11 +08:00
Zihao Ye 04522a76c6 [bugfix] Quick fix of #1547 (#1600)
* upd

* upd
2020-06-15 09:27:37 +08:00
Quan (Andy) Gan 57d111f9e5 [Bug] Fix broken initialization in Reddit (#1634) 2020-06-14 03:08:40 +08:00
zhjwy9343 02dd6d9021 [Feature] Add dot product attention (#1604)
* Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [New] Update landing page

* [New] Update landing page

* [New] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-06-12 09:29:39 +08:00
Mufei Li fe207b4564 [HeteroGraph] Improve Error Messages for find_edges (#1626)
* Improve error message

* Fix

* Fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-06-11 21:29:47 +08:00
Chao Ma a67ba946bc [RPC] Add send_request_to_machine() and remote_call_to_machine() (#1619)
* add send_request_to_machine()

* update

* update

* update

* update

* update

* update

* fix lint

* update
2020-06-11 13:04:06 +08:00
Da Zheng 50962c433f Refactor distributed graph store (#1597)
* refactor graph store for new kvstore.

* fix kvstore.

* fix.

* fix lint complains.

* add docstring.

* small fix in graph partition book.

* fix.

* remove tests

* disable sampling.

* Revert "disable sampling."

This reverts commit 70451008f61ff1481d5dadbf8bd199470aee559d.

* Revert "remove tests"

This reverts commit 1394364243bdd73b669abde6193a74e2cda5521d.
2020-06-10 18:51:32 -07:00
Jinjing Zhou 484bbcc8f1 [Test] Fix partition ci fail (#1614) 2020-06-10 15:44:38 +08:00
Chao Ma 8eab08d021 [KVStore] Remove Freeze flag (#1605)
* remove freeze

* update

* update

* fix lint
2020-06-09 15:32:02 +08:00
Mufei Li 36c7b77167 [LifeSci] Move to Independent Repo (#1592)
* Move LifeSci

* Remove doc
2020-06-05 21:28:24 +08:00
Chao Ma 64f49703af [KVStore] Re-write kvstore using DGL RPC infrastructure (#1569)
* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update init_data

* update server_state

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* debug init_data

* update

* update

* update

* update

* update

* update

* test get_meta_data

* update

* update

* update

* update

* update

* debug push

* update

* update

* update

* update

* update

* update

* update

* update

* update

* use F.reverse_data_type_dict

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* 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

* update

* fix lint

* update

* update

* update

* update

* fix test

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* set random seed

* update
2020-06-04 16:48:53 +08:00
Quan (Andy) Gan 2b9d06b6e2 [Data] Retry downloading once if dataset initialization fails (#1520)
* dataloader guard

* more fixes

* sets overwrite to true by default

* Update utils.py

* fix

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2020-06-03 17:21:05 +08:00
Mufei Li 90d2118da5 [Dataset] Change Cora Split (#1583)
* Use new version of Cora

* Fix import
2020-06-03 15:45:38 +08:00
Mufei Li a304df5d27 [Warning Message] Enrich Warning Information for Readonly Graphs (#1578)
* Enrich warning information for readonly graphs

* CI

* CI
2020-06-02 19:11:05 +08:00
Mufei Li a936f9d9a6 [Hetero] Allow Batching Graphs with Zero Nodes/Edges (#1575)
* Allow batching graphs with zero nodes and edges

* Fix the case without node/edge features

* Fix lint
2020-06-02 13:08:16 +08:00
Chao Ma 6a363378d5 [RPC] Init server_state during get_server_state() (#1559)
* init server_state

* update

* update
2020-05-25 15:41:45 +08:00
Chao Ma 3e69692233 [RPC] New RPC infrastructure. (#1549)
* WIP: rpc components

* client & server

* move network package to rpc

* fix include

* fix compile

* c api

* wip: test

* add basic tests

* missing file

* [RPC] Zero copy serializer (#1517)

* zerocopy serialization

* add test for HeteroGraph

* fix lint

* remove unnecessary codes

* add comment

* lint

* lint

* disable pylint for now

* add include for win

* windows guard

* lint

* lint

* skip test on windows

* refactor

* add comment

* fix

* comment

* 1111

* fix

* Update Jenkinsfile

* [RPC] Implementation of RPC infra (#1544)

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* remove client.cc and server.cc

* fix lint

* update

* update

* fix linr

* update

* fix lint

* update

* update

* update

* update

* update

* update

* update test

* update

* update test

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update comment

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* 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

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-05-22 18:48:37 +08:00
Quan (Andy) Gan 7ba9cbc63e [Docs] Fix docstrings and documentations (#1420)
* fix docs of to_block

* fix

* oops

* more fixes
2020-05-19 13:04:55 +08:00
Zihao Ye 70dc2ee9c6 [Feature] Add a request_format api and enrich related docstring. (#1528)
* upd

* upd

* lint

* fix

* bloody lint
2020-05-16 01:17:06 +08:00
Da Zheng 96984faca4 [Feature] Split data for distributed training (#1505)
* fix tests in graph partition book.

* implement node_split and edge_split.

* fix a bug.

* add tests.

* remove unnecessary code.

* avoid storing graph structure in partition book.

* add partition book in DistGraph.

* Revert "avoid storing graph structure in partition book."

This reverts commit 55b11fbf9293efcae5f8c97e93f1e121399dadae.

* small fixes.

* add a test for boolean mask vector.

* fix test.

* fix bugs.

* make it work for all different cases.

* fix tests.

* fix boolean mask

* fix for TF

* fix for tensorflow.

* fix test for TF

* only support boolean mask for now.

* fix tests.

* make the code more readable.

* fix test.

Co-authored-by: Chao Ma <mctt90@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-05-14 12:46:37 -07:00
Zihao Ye cfb24790ba [hotfix] Several bug fix for remove nodes/edges in DGLGraph. (#1521)
* upd

* upd

* better
2020-05-15 02:29:44 +08:00
Quan (Andy) Gan 20ec7bb0d9 [Bug?] Fixing compatibility of pickles across DGL versions (#1507)
* pickle compatibility before 0.4.2

* fixing for utils.Index

* add TODOs

* fix

* fix

* more compatibility checks for DGLGraph and DGLHeteroGraph objects

* lint

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-05-15 00:02:58 +08:00
Zihao Ye c2e61ce10b [Refactor] Restrict sparse format for DGLHeteroGraph (#1474)
* upd

* simplify

* further simplify

* lint

* doc

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* lint

* rename format

* upd

* lint

* upd

* upd

* upd

* udp

* debug

* upd

* upd

* upd

* upd

* upd

* 无可厚非吧

* 中三边肥

* 你一定要喊吗

* lint

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* Update unit_graph.h

Co-authored-by: Zihao Ye <yzh119@192.168.0.110>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2020-05-14 22:11:12 +08:00
Da Zheng 59aefe6e31 [Doc] write warning for remove_edges. (#1509)
* write warning for remove_edges.

* fix.

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-05-13 18:26:59 +08:00
Quan (Andy) Gan e5b10b10ee [Performance] fixing performance on heterogeneous graph creation in specific cases (#1506)
* [Performance] fixing performance on heterogeneous graph creation in specific cases

* fix for mxnet
2020-05-13 17:15:06 +08:00
Jinjing Zhou 6d04885381 [Backend] Turn to official dlpack for Tensorflow (#1511)
* Turn to official dlpack

* fix

* fix
2020-05-11 21:16:25 +08:00
Mohammad Amin Khashkhashi Moghaddam 0285021235 Fix bug in GIN Dataset degree OHE logic (#1498)
In the previous code, the same vector was being assigned to all the nodes in each graph.
`:` was used incorrectly assuming that only one row will be selected for each column but all rows were selected for each column.
2020-05-09 15:11:41 +08:00
Zihao Ye 1af0d8064a [bugfix] Fix the behavior of dgl.batch when batch size is 1 (#1483)
* fix

* upd

* upd

* upd

* upd

* clone

* upd

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-05-08 16:12:12 +08:00
Chao Ma ef7e475051 [Distributed] Add GraphPartitionBook (#1496)
* Add graph-partition-book

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* udpate

* fix lint

* update

* update

* update

* update

* update

* update

* update

* skip test on GPU

* update

* update

* update

* update

* update

* update

* update

* fix lint

* update

* update

* update

* update

* fix unittest

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2020-05-08 14:52:12 +08:00
Jinjing Zhou dc8ca88e58 [Refactor] Explicit dtype for HeteroGraph (#1467)
* 111

* 111

* lint

* lint

* lint

* lint

* fix

* lint

* try

* fix

* lint

* lint

* test

* fix

* ttt

* test

* fix

* fix

* fix

* mxnet

* 111

* fix 64bits computation

* pylint

* roll back

* fix

* lint

* fix hetero_from_relations

* remove index_dtype in to_homo and to_hetero

* fix

* fix

* fix

* fix

* remove default

* fix

* lint

* fix

* fix error message

* fix error

* lint

* macro dispatch

* try

* lint

* remove nbits

* error message

* fix

* fix

* lint

* lint

* lint

* fix

* lint

* fix

* fix random walk

* lint

* lint

* fix

* fix

* fix

* lint

* fix

* lint

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2020-05-08 14:26:28 +08:00
Da Zheng 16561a2e6b [Test] Add tests for TensorFlow (#1501)
* add test.

* move test code.

* remvoe unnecessary test.

* fix.

* turn on tests for TF.

* Revert "move test code."

This reverts commit e7b4f36395b2121a7be030bd4364a704d0e357bf.

* fix.

* fix.

* skip test for tensorflow.

Co-authored-by: Chao Ma <mctt90@gmail.com>
2020-05-06 00:33:53 -07:00
Da Zheng 2190c39d67 [Feature] Distributed graph store (#1383)
* initial version from distributed training.

This is copied from multiprocessing training.

* modify for distributed training.

* it's runnable now.

* measure time in neighbor sampling.

* simplify neighbor sampling.

* fix a bug in distributed neighbor sampling.

* allow single-machine training.

* fix a bug.

* fix a bug.

* fix openmp.

* make some improvement.

* fix.

* add prepare in the sampler.

* prepare nodeflow async.

* fix a bug.

* get id.

* simplify the code.

* improve.

* fix partition.py

* fix the example.

* add more features.

* fix the example.

* allow one partition

* use distributed kvstore.

* do g2l map manually.

* fix commandline.

* a temp script to save reddit.

* fix pull_handler.

* add pytorch version.

* estimate the time for copying data.

* delete unused code.

* fix a bug.

* print id.

* fix a bug

* fix a bug

* fix a bug.

* remove redundent code.

* revert modify in sampler.

* fix temp script.

* remove pytorch version.

* fix.

* distributed training with pytorch.

* add distributed graph store.

* fix.

* add metis_partition_assignment.

* fix a few bugs in distributed graph store.

* fix test.

* fix bugs in distributed graph store.

* fix tests.

* remove code of defining DistGraphStore.

* fix partition.

* fix example.

* update run.sh.

* only read necessary node data.

* batching data fetch of multiple NodeFlows.

* simplify gcn.

* remove unnecessary code.

* use the new copy_from_kvstore.

* update training script.

* print time in graphsage.

* make distributed training runnable.

* use val_nid.

* fix train_sampling.

* add distributed training.

* add run.sh

* add more timing.

* fix a bug.

* save graph metadata when partition.

* create ndata and edata in distributed graph store.

* add timing in minibatch training of GraphSage.

* use pytorch distributed.

* add checks.

* fix a bug in global vs. local ids.

* remove fast pull

* fix a compile error.

* update and add new APIs.

* implement more methods in DistGraphStore.

* update more APIs.

* rename it to DistGraph.

* rename to DistTensor

* remove some unnecessary API.

* remove unnecessary files.

* revert changes in sampler.

* Revert "simplify gcn."

This reverts commit 0ed3a34ca714203a5b45240af71555d4227ce452.

* Revert "simplify neighbor sampling."

This reverts commit 551c72d20f05a029360ba97f312c7a7a578aacec.

* Revert "measure time in neighbor sampling."

This reverts commit 63ae80c7b402bb626e24acbbc8fdfe9fffd0bc64.

* Revert "add timing in minibatch training of GraphSage."

This reverts commit e59dc8957a414c7df5c316f51d78bce822bdef5e.

* Revert "fix train_sampling."

This reverts commit ea6aea9a4aabb8ba0ff63070aa51e7ca81536ad9.

* fix lint.

* add comments and small update.

* add more comments.

* add more unit tests and fix bugs.

* check the existence of shared-mem graph index.

* use new partitioned graph storage.

* fix bugs.

* print error in fast pull.

* fix lint

* fix a compile error.

* save absolute path after partitioning.

* small fixes in the example

* Revert "[kvstore] support any data type for init_data() (#1465)"

This reverts commit 87b6997bf2.

* fix a bug.

* disable evaluation.

* Revert "Revert "[kvstore] support any data type for init_data() (#1465)""

This reverts commit f5b8039c6326eb73bad8287db3d30d93175e5bee.

* support set and init data.

* support set and init data.

* Revert "Revert "[kvstore] support any data type for init_data() (#1465)""

This reverts commit f5b8039c6326eb73bad8287db3d30d93175e5bee.

* fix bugs.

* fix unit test.

* move to dgl.distributed.

* fix lint.

* fix lint.

* remove local_nids.

* fix lint.

* fix test.

* remove train_dist.

* revert train_sampling.

* rename funcs.

* address comments.

* address comments.

Use NodeDataView/EdgeDataView to keep track of data.

* address comments.

* address comments.

* revert.

* save data with DGL serializer.

* use the right way of getting shape.

* fix lint.

* address comments.

* address comments.

* fix an error in mxnet.

* address comments.

* add edge_map.

* add more test and fix bugs.

Co-authored-by: Zheng <dzzhen@186590dc80ff.ant.amazon.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-131.us-east-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-26-167.us-east-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-150.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-250.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-30-135.us-west-2.compute.internal>
2020-05-02 19:30:51 -07:00
Chao Ma 5fc334fc30 [KVStore] Fix bug of get_meta() API (#1491)
* fix get-shape

* update

* update

* update

* update

* update

* fix typo

* update

* update
2020-05-02 15:01:42 +08:00
Jinjing Zhou 30b8074af4 [Fix] Fix stream creation and empty dict in tensor serialization (#1489)
* add functions

* fix litn

* add unit test

* fix

* fix

* fix

* fix

* support empty dict

* simplify logic

* Update tensor_serialize.py
2020-05-02 14:24:03 +08:00
Minjie Wang f25bc176d0 [Hetero] Improve speed of several Hetero APIs (#1486)
* add clone function to frame

* add utest

* replace all local_var with local_scope

* fix utest

* avoid creating canonical types in __getitem__

* lint

* try another utest  appraoch for mx

* utest
2020-05-01 15:21:33 +08:00
Minjie Wang 3c4506e91e [Bugfix] Add bool data type to backend. (#1487)
* add bool to F.data_type_dict

* add utest

* skip bool test for mx
2020-04-29 20:08:36 +08:00
Mohammad Amin Khashkhashi Moghaddam f1e4f3786d Fix bug in using degree_as_nlabel for GIN dataset (#1451)
In order to make the `label2idx` dictionary, tensor values should be used and not the tensor itself.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-04-29 17:42:45 +08:00
Jinjing Zhou 1506560e1e [Data] Add utils to save dict of tensors (#1481)
* add functions

* fix litn

* add unit test

* fix

* fix
2020-04-27 12:26:41 -07:00
Chao Ma 100ddd067e [KVStore] Bug fix (#1480)
* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* update

* Update network.cc

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2020-04-27 09:42:06 -07:00
Jinjing Zhou aa17d78af7 Fix (#1478) 2020-04-27 20:02:34 +08:00
Chao Ma 27520bc515 [KVStore] add init_data() on client (#1466)
* add init_data on client

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update

* update

* update
2020-04-25 17:59:23 -07:00
Chao Ma 87b6997bf2 [kvstore] support any data type for init_data() (#1465)
* support any data type for init_data

* update

* update
2020-04-23 22:04:40 +08:00
Minjie Wang 6cd7c313f1 [Bugfix] Fix corrupted memory when using multi-processing (#1453)
* fix

* fix mapping array for empty node set
2020-04-23 18:37:32 +08:00
Chao Ma 5efd1c4ace [KVstore] Improve user infterface (#1454)
* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2020-04-20 15:50:16 -07:00
Chao Ma 338f24cfbc [KVstore] Fast-pull (#1446)
* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint
2020-04-16 23:14:54 +08:00
Mufei Li bd1e48a51e [Bug] Fix khop graph (#1433)
* Update

* Update

* Update
2020-04-10 02:18:45 +08:00
Adam J. Stewart aefc080e30 [Doc] Fix GraphConv math notation (#1430) 2020-04-09 11:31:31 +08:00