-
Notifications
You must be signed in to change notification settings - Fork 304
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
Moves more MG graph ETL to libcugraph and re-enables MG tests in CI #3941
Conversation
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
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
… for graph creation (SG and MG)
…memory footprint when graphs are large relative to GPU memory
…to branch-23.12_re-enable-mg-testing
/ok to test |
/ok to test |
Added a |
There was a problem hiding this 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.
/ok to test |
/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, |
There was a problem hiding this comment.
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.
There was a problem hiding this 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.
/ok to test |
/merge |
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