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 #2116

Merged
merged 1 commit into from
Jan 23, 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 change allows CAGRA search to have an arbitrarily large top-k, instead of being limited to 1024 like in the previous code.

This works by using the multi-kernel search path, and replacing the _cuann_find_topk code with the matrix::select_k code - which can handle large K values.

Authors:
  - Ben Frederickson (https://github.com/benfred)

Approvers:
  - Tamas Bela Feher (https://github.com/tfeher)

URL: #2097
@GPUtester GPUtester requested a review from a team as a code owner January 23, 2024 22:22
@GPUtester GPUtester merged commit c90cdfa into branch-24.04 Jan 23, 2024
18 checks passed
@GPUtester
Copy link
Contributor Author

SUCCESS - forward-merge complete.

@github-actions github-actions bot added the cpp label Jan 23, 2024
abc99lr pushed a commit to abc99lr/raft that referenced this pull request Jan 23, 2024
This reverts commit c90cdfa, reversing
changes made to 0586fc3.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants