-
Notifications
You must be signed in to change notification settings - Fork 540
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
[REVIEW] Replace UMAP functionality dependency on nvgraph with RAFT Spectral Clustering #2500
[REVIEW] Replace UMAP functionality dependency on nvgraph with RAFT Spectral Clustering #2500
Conversation
Please update the changelog in order to start CI tests. View the gpuCI docs here. |
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 in general. Now that we've modularized the spectral implementation, it would be nice to take this opportunity to make the cluster solver execution optional when all we need are the eigenvectors.
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 pending update of commit hash once corresponding spectral PR is merged on the RAFT side.
Would also be nice to see the no-op clustering strategy make its way into raft eventually. I'm okay keeping it here until someone else needs it though.
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 we can remove the NVgraph dependency from CMakeLists.txt, otherwise this looks great.
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 @aschaffer for this PR! Just a couple of minor queries.
using index_type = int; | ||
using value_type = T; | ||
|
||
raft::handle_t r_handle; |
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.
@divyegala JFYI. Assuming this PR goes first before yours, we'll have to update this logic to use the raft handle that comes inside our cumlHandle.
rerun tests. |
rerun tests |
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
@teju85 could you re-review/approve this PR? Its just missing your approval to be able to merge |
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.
This is in conjunction with Spectral Clustering functionality move to RAFT (rapidsai/raft#12) and cugraph corresponding refactoring (rapidsai/cugraph#980).