文件历史

31 次代码提交

作者 SHA1 备注 提交日期
maqy1995 63284c08bb [Example] fix print info (#2305) 2020-11-27 03:17:12 +00:00
Mufei Li ad6b215bbd [Doc and User Guide] Minor Fix (#2206)
* [Distributed Example] Check args.num_servers with DGL_NUM_SERVER (#2203)

* update

* update

* update

* Fix doc and user guide

* Update heterograph.py

Co-authored-by: Chao Ma <mctt90@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-13-9.us-west-2.compute.internal>
2020-09-17 02:59:55 +08:00
Chao Ma c53bf77452 [Distributed Example] Add --close_profiler option to train_dist.py (#2199)
* update

* update
2020-09-15 14:37:28 +08:00
Qidong Su 0ced22c237 fix (#2142)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-10-127.us-west-2.compute.internal>
Co-authored-by: Chao Ma <mctt90@gmail.com>
2020-09-15 14:19:16 +08:00
Chao Ma 1b2658b926 update (#2192) 2020-09-14 15:35:39 +08:00
Chao Ma 1aa25cb618 update (#2191) 2020-09-14 15:13:39 +08:00
Leo 55f3e92631 [Fix] use DistDataLoader instead of Pytorch’s DataLoader. (#2182)
Co-authored-by: hzliuchw <hzliuchw@corp.netease.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-09-12 20:08:39 +08:00
Chao Ma e8273a8510 update README.md (#2138) 2020-09-02 15:42:28 +08:00
Chao Ma 4327d7125d [Distributed] Check num_workers and num_samplers (#2108)
* check num_workers

* update

* update

* update

* update

* update

* update
2020-08-27 12:42:25 +08:00
Chao Ma 283b2cfb96 [Distributed] Fix issues of GraphSage to support running on GPU (#2084)
* fix issues on GPU

* update

* update

* update

* update

* update

* update

* update

* update

* update

Co-authored-by: Ma <manchao@38f9d3587685.ant.amazon.com>
2020-08-25 13:20:46 +08:00
Chao Ma eb9c067b08 [Distributed] Copy training scripts in copy_partitions.py (#2010)
* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2020-08-13 12:20:39 +08:00
Chao Ma 98c1117d00 update (#2007) 2020-08-12 16:07:15 +08:00
Chao Ma f5d8fa8476 [Distributed] Fix all arguments to the format of xx_xxx (#2005)
* update

* update
2020-08-12 14:45:30 +08:00
Da Zheng d1cf5c383e [Distributed] adjust various APIs. (#1993)
* rename get_data_size.

* remove g from DistTensor.

* remove g from DistEmbedding.

* clean up API of graph partition book.

* fix DistGraph

* fix lint.

* collect all part policies.

* fix.

* fix.

* support distributed sampler.

* remove partition.py
2020-08-11 20:22:00 -07:00
Da Zheng 4efa332044 [Distributed] Move server start code to initialize. (#2002)
* move server start code to initialize.

* fix.

* fix lint.

* fix examples.

* add more checks.
2020-08-11 16:58:18 -07:00
Chao Ma d340ea3aa8 [Distributed] Remove server_count from ip_config.txt (#1985)
* remove server_count from ip_config.txt

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* lint

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* Update dist_context.py

* fix lint.

* make it work for multiple spaces.

* update ip_config.txt.

* fix examples.

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* udpate

* update

* update

* update

* update

* update

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-08-11 17:41:48 +08:00
Da Zheng ee30b2aadd [Distributed] Fix standalone (#1974)
* fix tests.

* fix.

* remove a test.

* make code work in the standalone mode.

* fix example.

* more fix.

* make DistDataloader work with num_workers=0

* fix DistDataloader tests.

* fix.

* fix lint.

* fix cleanup.

* fix test

* remove unnecessary code.

* remove tests.

* fix.

* fix.

* fix.

* fix example

* fix.

* fix.

* fix launch script.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-08-10 15:49:28 -07:00
Da Zheng 4b8eaf205b [Distributed] move feature copy to samplers in distributed GraphSage. (#1975)
* fix example.

* move feature copy to sampler.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-08-08 09:19:09 -07:00
Jinjing Zhou 4f499c7ffb [Distributed] DistDataloader (#1901)
* 111

* 111

* fix

* 111

* fix

* 11

* fix

* lint

* Update __init__.py

* lint

* fix

* lint

* fix

* fix

* fix

* fix

* fix

* try fix

* try fix

* fix

* Revert "fix"

This reverts commit a0b954fd4e99b7df92b53db8334dcb583d6e1551.

* fixes.

* fix.

* fix test.

* fix exit.

* fix.

* fix

* fix

* lint

* lint

* lint

* fix

* Update .gitignore

* 111

* fix

* 111

* 111

* fff

* 1111

* 111

* 1325315

* ffff

* f???

* fff

* 1111

* 111

* fix

* 111

* asda

* 1111

* 11

* 123

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

* spawn

* 1231231

* up

* 111

* fix

* fix

* Revert "fix"

This reverts commit 7373f95312fdcaa36d2fc330bf242339e89c045d.

* fix

* fix

* 1111

* fix

* fix tests

* start kvclient as early as possible.

* lint

* fix test

* lint

* 1111

* fix

* fix

* 111

* fix

* fix

* 1

* fix

* fix

* lint

* fix

* lint

* lint

* remove quit

* fix

* lint

* fix

* fix several

* lint

* fix minor

* fix

* lint

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-08-05 17:43:17 +08:00
Da Zheng a4c931a9bf [Distributed] Support multiple servers (#1886)
* client init graph on the backup servers.

* fix.

* test multi-server.

* fix anonymous dist tensors.

* check #parts.

* fix init_data

* add multi-server multi-client tests.

* update tests in kvstore.

* fix.

* verify the loaded partition.

* fix a bug.

* fix lint.

* fix.

* fix example.

* fix rpc.

* fix pull/push handler for backup kvstore

* fix example readme.

* change ip.

* update docstring.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-08-02 19:50:34 -07:00
xiang song(charlie.song) 56bbf9cb6d [Distributed][Example] Distributed training for unsupervised graphsage (#1853)
* Standalone can run

* fix

* Add save

* Fix

* Fix

* Fix

* Fix

* debug

* test

* test

* Fix

* Fix

* log

* Fix

* fix

* Profile

* auto sync grad

* update

* add test for unsupervised dist training

* upd

* Fix lr

* Fix update

* sync

* fix

* Revert "fix"

This reverts commit d5caa7398b36125f6d6e2c742a95c6ff4298c9e9.

* Fix

* unsupervised

* Fix

* remove debug

* Add test case for dist_graph find_edges()

* Fix

* skip tensorflow test for find_edges

* Update readme

* remove some test

* upd

* Update partition_graph.py

Co-authored-by: Ubuntu <ubuntu@ip-172-31-68-185.ec2.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-24-210.ec2.internal>
Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2020-08-01 12:24:55 -07:00
Da Zheng 4be4b13424 [Distributed] add copy_partitions.py (#1866)
* fix bugs.

* eval on both vaidation and testing.

* add script.

* update.

* update launch.

* make train_dist.py independent.

* update readme.

* update readme.

* update readme.

* update readme.

* generate undirected graph.

* rename conf_file to part_config

* use rsync

* make train_dist independent.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-115.us-west-2.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2020-07-31 11:11:56 -07:00
Chao Ma bcb988bdea [Distributed] Small fix on launch script (#1867)
* update

* update

* update

* update
2020-07-27 13:44:14 +08:00
Da Zheng 562871e76b [Distributed] add distributed evaluation. (#1810)
* add eval.

* extend DistTensor.

* fix.

* add barrier.

* add more print.

* add more checks in kvstore.

* fix lint.

* get all neighbors for eval.

* reorganize.

* fix.

* fix.

* fix.

* fix test.

* add reuse_if_exist.

* add test for reuse_if_exist.

* fix lint.

* fix bugs.

* fix.

* print errors of tcp socket.

* support delete tensors.

* fix lint.

* fix

* fix example

Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
2020-07-22 00:40:14 -07:00
Chao Ma 5c92f6c218 [RPC] Rpc exit with explicit invocation (#1825)
* exit client

* update

* update

* update

* update

* update

* update

* update

* update test

* update

* update

* update

* update

* update

* update

* update

* update

* update
2020-07-20 14:56:26 +08:00
Chao Ma ca9d3216d0 [Distributed] Distributed launching script (#1772)
* 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 launch script.

Co-authored-by: Da Zheng <zhengda1936@gmail.com>
2020-07-16 10:20:57 -07:00
Da Zheng cda0abf75a [Distributed] add the standalone mode in DistGraph (#1800)
* add standalone mode

* add comments.

* add tests for sampling.

* fix.

* make the code to run the standalone mode

* fix

* fix

* fix readme.

* fix.

* fix test

Co-authored-by: Chao Ma <mctt90@gmail.com>
2020-07-14 23:46:25 -07:00
Da Zheng a1472bcff6 [Distributed] Run distributed graph server inside DGL (#1801)
* run dist server in dgl.

* fix bugs.

* fix example.

* check environment variables and fix lint.

* fix lint
2020-07-14 07:41:44 -07:00
Quan (Andy) Gan 168a88e570 [Sampling] NodeDataLoader for node classification (#1635)
* neighbor sampler data loader first commit

* more commit

* nodedataloader

* fix

* update RGCN example

* update OGB

* fixes

* fix minibatch RGCN crashing with self loop

* reverting gatconv test code

* fix

* change to new solution that doesn't require tf dataloader

* fix

* lint

* fix

* fixes

* change doc

* fix docstring

* docstring fixes

* return seeds and input nodes from data loader

* fixes

* fix test

* fix windows build problem

* add pytorch wrapper

* fixes

* add pytorch wrapper

* add unit test

* add -1 support to sample_neighbors & fix docstrings

* docstring fix

* lint

* add minibatch rgcn evaluations

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2020-07-02 16:31:05 +08:00
Da Zheng 2c88f7c0b0 [Distributed] Split data to ensure data locality for multiple clients (#1710)
* fix.

* fix tests.

* fix

* add tests.

* fix.

* have default rank.

* add comment.

* fix test.

* remove check

* simplify code.

* add test.

* split data evenly.

* simplify the distributed training code.

* add comments.

* add comments.
2020-07-01 13:03:40 -07:00
Da Zheng 02d3197407 [Distributed] Pytorch example of distributed GraphSage. (#1495)
* add train_dist.

* Fix sampling example.

* use distributed sampler.

* fix a bug in DistTensor.

* fix distributed training example.

* add graph partition.

* add command

* disable pytorch parallel.

* shutdown correctly.

* load diff graphs.

* add ip_config.txt.

* record timing for each step.

* use ogb

* add profiler.

* fix a bug.

* add train_dist.

* Fix sampling example.

* use distributed sampler.

* fix a bug in DistTensor.

* fix distributed training example.

* add graph partition.

* add command

* disable pytorch parallel.

* shutdown correctly.

* load diff graphs.

* add ip_config.txt.

* record timing for each step.

* use ogb

* add profiler.

* add Ips of the cluster.

* fix exit.

* support multiple clients.

* balance node types and edges.

* move code.

* remove run.sh

* Revert "support multiple clients."

* fix.

* update train_sampling.

* fix.

* fix

* remove run.sh

* update readme.

* update readme.

* use pytorch distributed.

* ensure all trainers run the same number of steps.

* Update README.md

Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-250.us-west-2.compute.internal>
2020-06-28 13:09:40 +08:00