1. 10 Aug, 2020 1 commit
    • Da Zheng's avatar
      Fix the performance issue of graph partitioning in new DGLGraph (#1934) · 729ff2ef
      Da Zheng authored
      
      
      * fix perf.
      
      * fix.
      
      * accelerate metis.
      
      * fix lint.
      
      * use gklib.
      
      * fix perf.
      
      * fix.
      
      * update metis.
      
      * update launch script
      
      * handle synchronized API.
      
      * fix.
      
      * fix example.
      
      * fix dataloader.
      
      * temp fix.
      
      * temp fix omp.
      
      * distinguish roles.
      
      * initialize iterator of DistDataloader correctly.
      
      * check the correctness of launch script.
      
      * move feature copy to sampler.
      
      * measure mem/network copy time.
      
      * remove
      
      * Revert "measure mem/network copy time."
      
      This reverts commit 86cefdc14b7815fcf5aad6496af912dba48e4aa6.
      
      * fix.
      
      * fix
      
      * fix.
      
      * fix cmake.
      
      * disable metis in windows.
      
      * disable metis tests in windows.
      
      * remove test for multigraph.
      
      * fix test.
      
      * fix.
      
      * fix cmake.
      
      * fix.
      
      * revert.
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-19-115.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
      729ff2ef
  2. 08 Aug, 2020 1 commit
    • Da Zheng's avatar
      [Distributed] Add roles (#1971) · 5454471f
      Da Zheng authored
      * distinguish roles.
      
      * add comments.
      
      * fix lint.
      
      * move roles to server_state.
      
      * fix text.
      
      * fix tests.
      
      * fix tests.
      
      * Revert "fix tests."
      
      This reverts commit 5baa136b872a4550d4e612bfb1dfe363d7814adf.
      5454471f
  3. 05 Aug, 2020 1 commit
    • Jinjing Zhou's avatar
      [Distributed] DistDataloader (#1901) · 4f499c7f
      Jinjing Zhou authored
      
      
      * 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: default avatarDa Zheng <zhengda1936@gmail.com>
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      4f499c7f
  4. 03 Aug, 2020 1 commit
    • Da Zheng's avatar
      [Distributed] Support multiple servers (#1886) · a4c931a9
      Da Zheng authored
      
      
      * 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: default avatarUbuntu <ubuntu@ip-172-31-19-1.us-west-2.compute.internal>
      a4c931a9
  5. 31 Jul, 2020 1 commit
  6. 29 Jul, 2020 1 commit
  7. 27 Jul, 2020 2 commits
  8. 26 Jul, 2020 1 commit
  9. 22 Jul, 2020 1 commit
  10. 20 Jul, 2020 1 commit
    • Chao Ma's avatar
      [RPC] Rpc exit with explicit invocation (#1825) · 5c92f6c2
      Chao Ma authored
      * exit client
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update test
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      
      * update
      5c92f6c2
  11. 15 Jul, 2020 1 commit
  12. 14 Jul, 2020 1 commit
  13. 03 Jul, 2020 2 commits
    • Da Zheng's avatar
      [Feature] add sparse embedding. (#1497) · cadcc1c2
      Da Zheng authored
      * add sparse embedding.
      
      * fix
      
      * add test.
      
      * man fixes.
      
      * many fixes
      
      * fix sparse emb.
      
      * fix.
      
      * fix lint.
      
      * fix lint.
      
      * fix kvstore.
      
      * expose DistTensor.
      
      * test sparse embeddings.
      
      * add attach_grad to the backends.
      
      * remove part_id
      
      * fix.
      
      * move backward computation.
      
      * move more computation to backend.
      
      * fix a bug when applying learning rate.
      
      * fix a few things.
      
      * fix a few things.
      
      * add docstring
      
      * fix.
      
      * apply no_grad.
      
      * fix tests.
      
      * fix for other frameworks.
      
      * add examples in docstring.
      cadcc1c2
    • Da Zheng's avatar
      [Test] Fix test of distributed graph (#1748) · 17701174
      Da Zheng authored
      * Revert "Serialize server and client start for dist_graph_store test (#1736)"
      
      This reverts commit da16ebf0.
      
      * sleep
      17701174
  14. 02 Jul, 2020 1 commit
  15. 01 Jul, 2020 1 commit
  16. 18 Jun, 2020 1 commit
  17. 15 Jun, 2020 1 commit
    • Da Zheng's avatar
      [Feature] Range partition (#1522) · 41349dce
      Da Zheng authored
      
      
      * add reorder immutable graph.
      
      * add python API.
      
      * add reorder for csr.
      
      * remove gk version.
      
      * fix
      
      * add cpp test.
      
      * bug fixes
      
      * fix tests.
      
      * fix bugs and add check
      
      * fix test.
      
      * add omp.
      
      * add comments.
      
      * add coo reorder.
      
      * fix a bug.
      
      * handle reorder for different graph structues.
      
      * fix lint.
      
      * fix.
      
      * add original ids.
      
      * reshuffle nodes before metis partition.
      
      * inner nodes are in contiguous Id range.
      
      * reshuffle nodes/edges when partitioning.
      
      * load partition return graph partition book.
      
      * use inner_node/inner_edges
      
      * add and test range partition book.
      
      * count inner_edge correctly.
      
      * fix lint.
      
      * fix lint.
      
      * fix lint.
      
      * fix errors.
      
      * fix errors.
      
      * fix for TF.
      
      * fix.
      
      * fix.
      
      * change docstring.
      
      * support logical and.
      
      * add comments.
      
      * avoid copy.
      
      * fix
      
      * update docstring.
      
      * fix a bug.
      
      * add range search.
      
      * fix
      
      * fix a bug.
      
      * add more tests.
      
      * load graph partition book.
      
      * support shared memory for range partition book.
      
      * fix a bug.
      
      * fix.
      
      * fix lint.
      
      * remove check
      
      * fix test.
      
      * remove num_nodes and num_edges
      
      * fix lint.
      
      * fix graph partition book.
      
      * address comments.
      
      * use makedirs.
      
      * fix compile
      Co-authored-by: default avatarxiang song(charlie.song) <classicxsong@gmail.com>
      Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
      41349dce
  18. 11 Jun, 2020 3 commits
  19. 14 May, 2020 1 commit
    • Da Zheng's avatar
      [Feature] Split data for distributed training (#1505) · 96984fac
      Da Zheng authored
      
      
      * 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: default avatarChao Ma <mctt90@gmail.com>
      Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
      96984fac
  20. 03 May, 2020 1 commit
    • Da Zheng's avatar
      [Feature] Distributed graph store (#1383) · 2190c39d
      Da Zheng authored
      * 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 87b6997b
      
      .
      
      * 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: default avatarZheng <dzzhen@186590dc80ff.ant.amazon.com>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-6-131.us-east-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-26-167.us-east-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-16-150.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-16-250.us-west-2.compute.internal>
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-30-135.us-west-2.compute.internal>
      2190c39d