-
Notifications
You must be signed in to change notification settings - Fork 309
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
Conversation
cpp/src/mtmg/vertex_result.cu
Outdated
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, |
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.
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.
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.
Updated in latest push.
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.
LGTM
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.
Looks good to me
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, It would be good start point to implement MTMG for other algorithms.
/merge |
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