Unverified Commit 41349dce authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

[Feature] Range partition (#1522)



* 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>
parent 3e72c53a
...@@ -109,7 +109,7 @@ def test_node_subgraph_with_halo(): ...@@ -109,7 +109,7 @@ def test_node_subgraph_with_halo():
inner_node = inner_node.asnumpy() inner_node = inner_node.asnumpy()
inner_node_ids = np.nonzero(inner_node)[0] inner_node_ids = np.nonzero(inner_node)[0]
inner_node_ids = halo_subg.induced_nodes.tonumpy()[inner_node_ids] inner_node_ids = halo_subg.induced_nodes.tonumpy()[inner_node_ids]
assert np.all(inner_node_ids == np.sort(nodes)) assert np.all(np.sort(inner_node_ids) == np.sort(nodes))
# Check if the inner edge labels are correct. # Check if the inner edge labels are correct.
inner_edge = inner_edge.asnumpy() inner_edge = inner_edge.asnumpy()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment