Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Moves more MG graph ETL to libcugraph and re-enables MG tests in CI #3941

Merged

Conversation

jnke2016
Copy link
Contributor

@jnke2016 jnke2016 commented Oct 17, 2023

This PR includes changes that moves some of the MG graph etl steps (such as computing number of edges) to libcugraph to reduce the amount of dask overhead involved in graph creation. Those ETL steps were also responsible for various dask-related transient errors that caused us to temporarily disable MG testing in CI. These changes allow us to re-enable MG testing in CI, so this PR includes that update too.

closes #3948

@jnke2016 jnke2016 requested a review from a team as a code owner October 17, 2023 18:07
@rlratzel rlratzel marked this pull request as draft October 17, 2023 20:12
rapids-bot bot pushed a commit that referenced this pull request Oct 17, 2023
This PR temporarily disables single-GPU "MG" tests in CI until the dask related transient failures are resolved.

The PR to re-enable them, intended to block the 23.12 release, is [here](#3941).

Authors:
  - Joseph Nke (https://github.com/jnke2016)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)
  - Ray Douglass (https://github.com/raydouglass)

URL: #3940
divyegala pushed a commit to divyegala/cugraph that referenced this pull request Oct 18, 2023
This PR temporarily disables single-GPU "MG" tests in CI until the dask related transient failures are resolved.

The PR to re-enable them, intended to block the 23.12 release, is [here](rapidsai#3941).

Authors:
  - Joseph Nke (https://github.com/jnke2016)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)
  - Ray Douglass (https://github.com/raydouglass)

URL: rapidsai#3940
@BradReesWork BradReesWork added this to the 23.12 milestone Nov 15, 2023
@jnke2016 jnke2016 marked this pull request as ready for review November 15, 2023 21:23
@jnke2016 jnke2016 requested review from a team as code owners November 15, 2023 21:23
@jnke2016 jnke2016 requested a review from a team as a code owner November 17, 2023 22:39
@jnke2016
Copy link
Contributor Author

/ok to test

@jnke2016
Copy link
Contributor Author

/ok to test

@rlratzel rlratzel changed the title Placeholder for re-enabling the mg testing Moves more MG graph ETL to libcugraph and re-enables MG tests in CI Nov 22, 2023
@rlratzel rlratzel added the DO NOT MERGE Hold off on merging; see PR for details label Nov 22, 2023
@rlratzel
Copy link
Contributor

Added a DO NOT MERGE label because Joseph has at least one more commit to push.

Copy link
Collaborator

@ChuckHastings ChuckHastings left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

C changes (updated documentation) look good.

@jnke2016
Copy link
Contributor Author

/ok to test

@jnke2016
Copy link
Contributor Author

/ok to test

weight_array=None,
store_transposed=False,
renumber=False,
do_expensive_check=False,
edge_id_array=None,
edge_type_array=None,
input_array_format="COO",
vertices_array=None,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This (and also the updated MGGraph ctor) still does not have all new args at the end, but maybe since only drop_self_loops and drop_multi_edges are affected, it's not that likely to break users using only positional params? I'll approve it but I'll also label this PR as a breaking change.

Copy link
Contributor

@rlratzel rlratzel left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Approved, but I'll label the PR as a breaking change since users constructing PLC graph objects that include drop_self_loops and drop_multi_edges as positional args will be affected.

@rlratzel rlratzel added breaking Breaking change and removed non-breaking Non-breaking change labels Nov 26, 2023
@jnke2016
Copy link
Contributor Author

/ok to test

@rlratzel rlratzel removed the DO NOT MERGE Hold off on merging; see PR for details label Nov 27, 2023
@rlratzel
Copy link
Contributor

/merge

@rapids-bot rapids-bot bot merged commit 3116eed into rapidsai:branch-23.12 Nov 27, 2023
73 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
breaking Breaking change improvement Improvement / enhancement to an existing function
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Update python/dask cugraph implementation to use new PLC features from #3947
7 participants