Skip to content

Add float16 support for CAGRA extend#1620

Merged
rapids-bot[bot] merged 10 commits intorapidsai:mainfrom
lowener:26.02-cagra-extend
Jan 28, 2026
Merged

Add float16 support for CAGRA extend#1620
rapids-bot[bot] merged 10 commits intorapidsai:mainfrom
lowener:26.02-cagra-extend

Conversation

@lowener
Copy link
Copy Markdown
Contributor

@lowener lowener commented Dec 8, 2025

Related issue: #1595
CAGRA support extend for float32, int8, uint8. This PR adds float16. I conserved the commits from the original authors and added some tests on C++ and python sides.

@lowener lowener added the bug Something isn't working label Dec 8, 2025
@lowener lowener requested a review from a team as a code owner December 8, 2025 15:35
@lowener lowener added the non-breaking Introduces a non-breaking change label Dec 8, 2025
@lowener lowener requested review from a team as code owners December 8, 2025 15:35
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Dec 8, 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.

@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Dec 8, 2025

/ok to test 46d53b7

@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Dec 15, 2025

/ok to test d2e7798

@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Dec 22, 2025

/ok to test 8ba5eed

@cjnolet
Copy link
Copy Markdown
Member

cjnolet commented Jan 5, 2026

Thanks for spotting and fixing this @lowener. Just for completeness, can you share the increase in binary size that this change causes? Likely not. a blocker, but helps us for tracking to keep that info close to the changes. Thanks!

@cjnolet cjnolet moved this from Todo to In Progress in Unstructured Data Processing Jan 5, 2026
@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Jan 26, 2026

/ok to test bcc7d5b

Signed-off-by: Mickael Ide <[email protected]>
Copy link
Copy Markdown
Member

@dantegd dantegd left a comment

Choose a reason for hiding this comment

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

One docstring comment, other than that lgtm

Comment thread cpp/include/cuvs/neighbors/cagra.hpp Outdated
Comment thread cpp/include/cuvs/neighbors/cagra.hpp Outdated
Comment thread cpp/include/cuvs/neighbors/cagra.hpp Outdated
Co-authored-by: Dante Gama Dessavre <[email protected]>
@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Jan 26, 2026

/ok to test dc3f333

@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Jan 28, 2026

/ok to test ed4646f

@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Jan 28, 2026

/merge

@rapids-bot rapids-bot Bot merged commit 5ab16ea into rapidsai:main Jan 28, 2026
91 checks passed
@github-project-automation github-project-automation Bot moved this from In Progress to Done in Unstructured Data Processing Jan 28, 2026
@lowener
Copy link
Copy Markdown
Contributor Author

lowener commented Jan 28, 2026

Increase in binary size for all architecture: libcuvs.so went from 1096M to 1098M. So +2M

@lowener lowener deleted the 26.02-cagra-extend branch January 28, 2026 13:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working non-breaking Introduces a non-breaking change

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

4 participants