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

Add support for Louvain to MTMG #4033

Merged
merged 7 commits into from
Dec 6, 2023

Conversation

ChuckHastings
Copy link
Collaborator

@ChuckHastings ChuckHastings commented Dec 1, 2023

Added vertex result instantiation for vertex_t which is necessary to handle Louvain results.

Added an MTMG test for Louvain to demonstrate how to use Louvain in MTMG.

Closes https://github.com/rapidsai/graph_dl/issues/330

@ChuckHastings ChuckHastings added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Dec 1, 2023
@ChuckHastings ChuckHastings self-assigned this Dec 1, 2023
@ChuckHastings ChuckHastings added this to the 24.02 milestone Dec 1, 2023
@ChuckHastings ChuckHastings marked this pull request as ready for review December 2, 2023 02:41
@ChuckHastings ChuckHastings requested review from a team as code owners December 2, 2023 02:41
template rmm::device_uvector<float> vertex_result_view_t<float>::gather(
handle_t const& handle,
raft::device_span<int32_t const> vertices,
cugraph::mtmg::graph_view_t<int32_t, int32_t, false, false> const& graph_view,
Copy link
Contributor

Choose a reason for hiding this comment

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

Should we really take graph_view here or we can just take vertex partitioning ranges? Many of these explicit instantiations won't be necessary if we just take something like vertex_partition_range_lasts (https://github.com/rapidsai/cugraph/blob/branch-24.02/cpp/include/cugraph/graph_functions.hpp#L237).

LGTM otherwise.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Updated in latest push.

@github-actions github-actions bot added doc Documentation cuGraph CMake labels Dec 4, 2023
Copy link
Contributor

@seunghwak seunghwak left a comment

Choose a reason for hiding this comment

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

LGTM

@ChuckHastings ChuckHastings removed the doc Documentation label Dec 5, 2023
Copy link
Contributor

@jnke2016 jnke2016 left a comment

Choose a reason for hiding this comment

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

Looks good to me

Copy link
Contributor

@naimnv naimnv left a comment

Choose a reason for hiding this comment

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

Thanks, It would be good start point to implement MTMG for other algorithms.

@ChuckHastings
Copy link
Collaborator Author

/merge

@rapids-bot rapids-bot bot merged commit f80480d into rapidsai:branch-24.02 Dec 6, 2023
74 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CMake cuGraph improvement Improvement / enhancement to an existing function non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants