Skip to content

Fix pq_compressed file created by SSD Vamana build#1313

Merged
rapids-bot[bot] merged 4 commits intorapidsai:branch-25.10from
bkarsin:vamana_pq_fix
Sep 16, 2025
Merged

Fix pq_compressed file created by SSD Vamana build#1313
rapids-bot[bot] merged 4 commits intorapidsai:branch-25.10from
bkarsin:vamana_pq_fix

Conversation

@bkarsin
Copy link
Copy Markdown
Contributor

@bkarsin bkarsin commented Sep 4, 2025

Fix issue with pq_compressed files being incorrect when GPU vamana build was being provided opq codebooks as an input to the build. Bug was caused by vq_centers not being set to 0s, despite OPQ compression not using vq centers, and only using pq_centers during quantized vector encoding.

Also cleaned up vamana_example file to support other datatypes and function correctly when building the DiskANN index without quantization.

Comment thread cpp/src/neighbors/detail/vamana/vamana_build.cuh Outdated
@bkarsin bkarsin added the bug Something isn't working label Sep 5, 2025
@cjnolet cjnolet added the non-breaking Introduces a non-breaking change label Sep 5, 2025
@tarang-jain
Copy link
Copy Markdown
Contributor

There are some merge conflicts. Other than that this PR looks good!

@cjnolet cjnolet moved this from Todo to In Progress in Unstructured Data Processing Sep 15, 2025
@cjnolet
Copy link
Copy Markdown
Member

cjnolet commented Sep 16, 2025

/merge

@rapids-bot rapids-bot Bot merged commit 703a102 into rapidsai:branch-25.10 Sep 16, 2025
84 checks passed
@github-project-automation github-project-automation Bot moved this from In Progress to Done in Unstructured Data Processing Sep 16, 2025
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.

3 participants