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

Forward-merge branch-24.02 to branch-24.04 #2122

Merged
merged 1 commit into from
Jan 24, 2024
Merged

Conversation

GPUtester
Copy link
Contributor

Forward-merge triggered by push to branch-24.02 that creates a PR to keep branch-24.04 up-to-date. If this PR is unable to be immediately merged due to conflicts, it will remain open for the team to manually merge.

This PR adds support for an eps-neighborhood search based on random ball cover. The algorithm re-uses the existing rbc index creation.

Changes:
* add C++ API `raft::neighbors::ball_cover::epsUnexpL2NeighborhoodRbc`
* lifted the 2/3-D limitation for eps-neighborhod via RBC (limitation still in place for k-nn queries)
* pylibraft support for dense brute-force eps-neighborhood
* pylibraft support for sparse/dense rbc epsneighborhood

Note: The PR also contains a fix for the vertex degree computation in the brute force algorithm `spatial::knn::detail::epsUnexpL2SqNeighborhood`.

Related to #1984 and #517

Authors:
  - Malte Förster (https://github.com/mfoerste4)
  - Tamas Bela Feher (https://github.com/tfeher)
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Tamas Bela Feher (https://github.com/tfeher)
  - Corey J. Nolet (https://github.com/cjnolet)

URL: #2028
@GPUtester GPUtester requested review from a team as code owners January 24, 2024 02:32
@GPUtester GPUtester merged commit 2647757 into branch-24.04 Jan 24, 2024
17 of 18 checks passed
@GPUtester
Copy link
Contributor Author

SUCCESS - forward-merge complete.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants