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 brute_force index serialization #2036

Merged
merged 22 commits into from
Dec 7, 2023

Conversation

wphicks
Copy link
Contributor

@wphicks wphicks commented Dec 3, 2023

Add serialization and deserialization methods for brute_force index. Also add overloads to brute_force search and build functions taking index_param and search_param arguments for API compatibility with other index types.

@wphicks wphicks requested review from a team as code owners December 3, 2023 22:00
Copy link
Member

@benfred benfred 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 this @wphicks ! It looks great,

cpp/include/raft/neighbors/brute_force_serialize.cuh Outdated Show resolved Hide resolved
cpp/test/neighbors/ann_brute_force.cuh Outdated Show resolved Hide resolved
@wphicks
Copy link
Contributor Author

wphicks commented Dec 4, 2023

Thanks @benfred! Agreed with all the suggestions and updated accordingly.

Copy link
Member

@benfred benfred left a comment

Choose a reason for hiding this comment

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

lgtm! thanks @wphicks

Copy link
Member

@cjnolet cjnolet left a comment

Choose a reason for hiding this comment

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

I think this looks really good and we've needed this! Just one thing- we've addressed this in other indices which require the original dataset available and I'd like to use the same design pattern here so that the dataset doesn't need to always be serialized out with the norms.

@wphicks wphicks requested a review from cjnolet December 5, 2023 18:29
@wphicks wphicks added 4 - Waiting on Reviewer Waiting for reviewer to review or respond and removed 3 - Ready for Review labels Dec 5, 2023
Copy link
Member

@cjnolet cjnolet left a comment

Choose a reason for hiding this comment

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

LGTM!

@cjnolet
Copy link
Member

cjnolet commented Dec 7, 2023

I think there's a small failure in the docs build, but otherwise it looks great!

@wphicks
Copy link
Contributor Author

wphicks commented Dec 7, 2023

Aha! Forgot to add include_dataset parameter to the docs after I implemented it. Just pushed the update. Many thanks for the reviews, @benfred and @cjnolet!

@wphicks
Copy link
Contributor Author

wphicks commented Dec 7, 2023

/merge

@rapids-bot rapids-bot bot merged commit d2210a2 into rapidsai:branch-24.02 Dec 7, 2023
61 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
4 - Waiting on Reviewer Waiting for reviewer to review or respond CMake cpp feature request New feature or request non-breaking Non-breaking change
Projects
Development

Successfully merging this pull request may close these issues.

3 participants