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

Calculate max cluster size correctly for IVF-PQ #938

Merged

Conversation

tfeher
Copy link
Contributor

@tfeher tfeher commented Oct 21, 2022

This PR fixes the calculation of max cluster size.

Previous calculation could return much larger than the actual values, which could lead to OOM while allocating temporary buffers while building index for large datasets.

@tfeher tfeher requested a review from achirkin October 21, 2022 15:30
@tfeher tfeher requested a review from a team as a code owner October 21, 2022 15:30
@github-actions github-actions bot added the cpp label Oct 21, 2022
@cjnolet cjnolet added improvement Improvement / enhancement to an existing function non-breaking Non-breaking change labels Oct 21, 2022
Copy link
Contributor

@achirkin achirkin 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 getting to the root of this, Tamas! Looks good.

cpp/include/raft/spatial/knn/detail/ivf_pq_build.cuh Outdated Show resolved Hide resolved
cpp/include/raft/spatial/knn/detail/ivf_pq_build.cuh Outdated Show resolved Hide resolved
@tfeher
Copy link
Contributor Author

tfeher commented Oct 24, 2022

Thanks @achirkin for the review, I have addressed the issue.

@cjnolet
Copy link
Member

cjnolet commented Oct 24, 2022

@gpucibot merge

@rapids-bot rapids-bot bot merged commit 73cd988 into rapidsai:branch-22.12 Oct 24, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cpp improvement Improvement / enhancement to an existing function non-breaking Non-breaking change
Projects
Development

Successfully merging this pull request may close these issues.

3 participants