[GraphBolt] Allow using multiple processes for GraphBolt partition conversion

thvasilo commented 3 months ago

Description

Add option to use multi-processing to speed up conversion of DGL to GraphBolt partitions.

Checklist

Please feel free to remove inapplicable items for your PR.

[ ] The PR title starts with [$CATEGORY] (such as [NN], [Model], [Doc], [Feature]])
[ ] I've leverage the tools to beautify the python and c++ code.
[ ] The PR is complete and small, read the Google eng practice (CL equals to PR) to understand more about small PR. In DGL, we consider PRs with less than 200 lines of core code change are small (example, test and documentation could be exempted).
[ ] All changes have test coverage
[ ] Code is well-documented
[ ] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change
[ ] Related issue is referred in this PR
[ ] If the PR is for a new model/paper, I've updated the example index here.

Changes

dgl-bot commented 3 months ago

To trigger regression tests:

@dgl-bot run [instance-type] [which tests] [compare-with-branch]; For example: @dgl-bot run g4dn.4xlarge all dmlc/master or @dgl-bot run c5.9xlarge kernel,api dmlc/master

dgl-bot commented 3 months ago

Commit ID: 644ad13a34ddc91aa20c21c5848d6c29d17fcf61

Build ID: 1

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: e44aecddf1196c1453e2897f2e26742e5a21d9e9

Build ID: 2

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 05a6da3d9c3c50049ddb826994aacd4ca9c96675

Build ID: 3

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 4e150ecca541e532b2aa14ef7ec55d43441bf47e

Build ID: 4

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

thvasilo commented 2 months ago

@Rhett-Ying updated to use Python libs for mp

dgl-bot commented 2 months ago

Commit ID: 17ba4689649310c7f1aa0e58b6eeb98059874748

Build ID: 5

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 1daf5c2c0d483d42a74077b499046d59dd478107

Build ID: 6

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 61aaff2944c33570e847748deee9ef4dad31f4c1

Build ID: 7

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: bab12decdc0ba55942ae84441952e3f13c2a927e

Build ID: 8

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 8af82daaab829a5fade91f0c1d345a822117bb47

Build ID: 9

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 2576900ee362b7b44b2044183d4941c05c612373

Build ID: 10

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: bc1e7a3a600571d9fc498a4579e2a206ba40b8ed

Build ID: 11

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

thvasilo commented 2 months ago

@Rhett-Ying could you take another look? I've had to use spawn for the multi-processing, otherwise the processes would hang. My guess is that it was happening because of the CAPI calls not acquiring/releasing the GIL correctly?

Rhett-Ying commented 2 months ago

please fix the lint check error

dgl-bot commented 2 months ago

Commit ID: 32b50b33605631f512622e4b5bb1b6b840e86708

Build ID: 12

Status: ❌ CI test failed in Stage [Lint Check].

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 06e59b8fb1dc4c63e0596c3577bc6d4ce43b6f90

Build ID: 13

Status: ⚪️ CI test cancelled due to overrun.

Report path: link

Full logs path: link

dgl-bot commented 2 months ago

Commit ID: 97cba89b06228dbd21dba92cfd4cb2061ca6ff22

Build ID: 14

Status: ✅ CI test succeeded.

Report path: link

Full logs path: link

thvasilo commented 2 months ago

Hi @Rhett-Ying CI passes now, let me know if there's any work left here.

dmlc / dgl