Skip to content

Add support for half in CAGRA+HNSW #813

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

Merged
merged 16 commits into from
May 7, 2025

Conversation

lowener
Copy link
Contributor

@lowener lowener commented Apr 8, 2025

This PR add support for half dtype for HNSW in C++, C and python, as well as some tests with it. I had to modify a bit the HNSW patch in order to enable computation on half data types.
I also added the support of inner-product distance for int8/uint8 data type.

Signed-off-by: Mickael Ide <[email protected]>
@lowener lowener added non-breaking Introduces a non-breaking change C++ feature request New feature or request labels Apr 8, 2025
@lowener lowener requested a review from a team as a code owner April 8, 2025 14:54
@lowener lowener requested a review from raydouglass April 8, 2025 14:54
Copy link
Member

@divyegala divyegala left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for your improvement to the docs! Looks like you caught a lot of typos.

@lowener lowener requested a review from divyegala April 14, 2025 11:20
@divyegala divyegala removed request for a team and raydouglass April 14, 2025 18:12
Signed-off-by: Mickael Ide <[email protected]>
Copy link

copy-pr-bot bot commented Apr 17, 2025

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.

@rapidsai rapidsai deleted a comment from copy-pr-bot bot Apr 18, 2025
@lowener
Copy link
Contributor Author

lowener commented Apr 18, 2025

/ok to test 6660dff

@lowener lowener requested a review from divyegala April 24, 2025 13:10
Signed-off-by: Mickael Ide <[email protected]>
@rapidsai rapidsai deleted a comment from copy-pr-bot bot Apr 24, 2025
@lowener
Copy link
Contributor Author

lowener commented Apr 24, 2025

/ok to test c6a93c2

@cjnolet
Copy link
Member

cjnolet commented Apr 25, 2025

/ok to test 72790a4

Copy link
Member

@divyegala divyegala left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pre-approving pending 2 comments. Everything else LGTM!

@lowener
Copy link
Contributor Author

lowener commented Apr 28, 2025

/ok to test c3c82e1

Signed-off-by: Mickael Ide <[email protected]>
@lowener
Copy link
Contributor Author

lowener commented Apr 30, 2025

/ok to test 8f51529

@divyegala
Copy link
Member

/ok to test c7c56b5

Signed-off-by: Mickael Ide <[email protected]>
@lowener
Copy link
Contributor Author

lowener commented May 6, 2025

/ok to test eab0cbe

@cjnolet
Copy link
Member

cjnolet commented May 7, 2025

/merge

@rapids-bot rapids-bot bot merged commit 4299c98 into rapidsai:branch-25.06 May 7, 2025
66 checks passed
@lowener lowener deleted the 25.06-cagra-half branch May 8, 2025 12:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
C++ CMake cpp feature request New feature or request non-breaking Introduces a non-breaking change Python
Development

Successfully merging this pull request may close these issues.

3 participants