Skip to content

fix knn_merge_parts for inner product - #2426

Open
qwertyforce wants to merge 2 commits into
NVIDIA:mainfrom
qwertyforce:knn_merge_parts_fix_ip
Open

fix knn_merge_parts for inner product#2426
qwertyforce wants to merge 2 commits into
NVIDIA:mainfrom
qwertyforce:knn_merge_parts_fix_ip

Conversation

@qwertyforce

Copy link
Copy Markdown
Contributor

By default, when merging the results of sharded multigpu index, knn_merge_parts keeps K smallest values. But inner product is not a distance, it is a measure of similarity. Therefore results are wrong for IP.
In this PR we are adding an additional overload, that receives an argument select_min, we keep backward compatibility and add a new test

@copy-pr-bot

copy-pr-bot Bot commented Aug 7, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant