-
Notifications
You must be signed in to change notification settings - Fork 68
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
Conversation
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] |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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.
/merge |
Authors: - Ben Frederickson (https://github.com/benfred) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: rapidsai/cuvs#175
No description provided.