-
Notifications
You must be signed in to change notification settings - Fork 310
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
Enable parallel mode #3875
Enable parallel mode #3875
Conversation
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.
Some C API comments to start with.
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.
Thanks, Joseph, I'm looking forward to having this capability. I found a few issues below.
fixture_params = gen_fixture_params_product( | ||
(datasets, "graph_file"), | ||
([True, False], "distributed"), | ||
([True, False], "use_weights"), | ||
([True, False], "use_edge_ids"), | ||
([True, False], "use_edge_type_ids"), | ||
) | ||
|
||
|
||
@pytest.fixture(scope="module", params=fixture_params) | ||
def input_combo(request): | ||
""" | ||
Simply return the current combination of params as a dictionary for use in | ||
tests or other parameterized fixtures. | ||
""" | ||
return dict( | ||
zip( | ||
( | ||
"graph_file", | ||
"use_weights", | ||
"use_edge_ids", | ||
"use_edge_type_ids", | ||
"distributed", | ||
), | ||
request.param, | ||
) | ||
) |
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 pattern isn't as useful here since there's only one test. It might be better for future maintainers if the test in this file just used regular pytest parameters.
48e3f00
to
58014b9
Compare
0146339
to
3f09595
Compare
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.
I think I found out why CI is failing for this PR.
python/cugraph/cugraph/tests/internals/test_replicate_edgelist_mg.py
Outdated
Show resolved
Hide resolved
/merge |
This PR enables parallel mode
Closes https://github.com/rapidsai/graph_dl/issues/328