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

Add refine to the Python and C api's #175

Merged
merged 5 commits into from
Jun 7, 2024

Conversation

benfred
Copy link
Member

@benfred benfred commented Jun 5, 2024

No description provided.

@benfred benfred added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Jun 5, 2024
@benfred benfred requested review from a team as code owners June 5, 2024 00:17
dataset : array interface compliant matrix, shape (n_samples, dim)
Supported dtype [float, int8, uint8]
queries : array interface compliant matrix, shape (n_queries, dim)
Supported dtype [float, int8, uint8]
Copy link
Contributor

Choose a reason for hiding this comment

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

Float16 is also supported? Or not in python?

Copy link
Member Author

Choose a reason for hiding this comment

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

Thats a good callout! float16 wasn't supported in the python api (and wasn't in the raft python refine api either) - but was supported in the c++ api.

I added support for float16 in the python refine api in the last commit.

@benfred benfred linked an issue Jun 6, 2024 that may be closed by this pull request
@cjnolet
Copy link
Member

cjnolet commented Jun 7, 2024

/merge

@rapids-bot rapids-bot bot merged commit b771085 into rapidsai:branch-24.08 Jun 7, 2024
54 checks passed
@benfred benfred deleted the refine_python branch June 7, 2024 05:00
difyrrwrzd added a commit to difyrrwrzd/cuvs that referenced this pull request Aug 10, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CMake cpp improvement Improves an existing functionality non-breaking Introduces a non-breaking change Python
Projects
Development

Successfully merging this pull request may close these issues.

[FEA] Expose refine APIs through Python
3 participants