Skip to content

[ANN_BENCH] Fix OpenMP Sections in Wrappers #911

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

Open
wants to merge 18 commits into
base: branch-25.08
Choose a base branch
from

Conversation

tarang-jain
Copy link
Contributor

Fixes OpenMP num thread setting + dispatching search workloads in DiskANN and Faiss CPU cuvs-bench wrappers.
Hnswlib wrapper taken care of in #724

@tarang-jain tarang-jain self-assigned this May 20, 2025
Copy link

copy-pr-bot bot commented May 20, 2025

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@github-actions github-actions bot added the cpp label May 20, 2025
@tarang-jain tarang-jain added bug Something isn't working non-breaking Introduces a non-breaking change benchmarking labels May 20, 2025
@tarang-jain tarang-jain marked this pull request as ready for review May 22, 2025 19:18
@tarang-jain tarang-jain requested a review from a team as a code owner May 22, 2025 19:18
@tarang-jain
Copy link
Contributor Author

@achirkin @tfeher for viz. Removed number of search threads entirely because user can control it with OpenMP environment variables, which is actually the most correct way of using OpenMP rather than passing in process-specific threads through cuvs-bench. We simply call Faiss' search APIs with the batch of queries and Faiss does all its parallelization internally.

@tarang-jain tarang-jain changed the base branch from branch-25.06 to branch-25.08 June 1, 2025 11:19
Copy link
Member

@divyegala divyegala left a comment

Choose a reason for hiding this comment

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

@tarang-jain can you pull #724 into this PR?

@tarang-jain
Copy link
Contributor Author

@divyegala #724 has build failures in CI at the moment.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
benchmarking bug Something isn't working cpp non-breaking Introduces a non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants