Skip to content

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-pr-bot
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.

@divyegala
Copy link
Member

/merge

@rapids-bot rapids-bot bot merged commit f17ada8 into rapidsai:branch-25.08 Jul 1, 2025
53 checks passed
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

Development

Successfully merging this pull request may close these issues.

3 participants