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

FAISS with cuVS enabled in cuvs-bench #561

Open
wants to merge 22 commits into
base: branch-25.02
Choose a base branch
from

Conversation

tarang-jain
Copy link
Contributor

Mostly adapted from rapidsai/raft#2026

Copy link

copy-pr-bot bot commented Jan 8, 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.

@tarang-jain tarang-jain added non-breaking Introduces a non-breaking change improvement Improves an existing functionality labels Jan 9, 2025
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.

Here are couple suggestions on my side to keep the faiss behavior in line with other algorithms.

cpp/bench/ann/src/faiss/faiss_gpu_benchmark.cu Outdated Show resolved Hide resolved
@tarang-jain tarang-jain marked this pull request as ready for review January 22, 2025 17:19
@tarang-jain tarang-jain requested review from a team as code owners January 22, 2025 17:19
@@ -484,6 +484,7 @@ void register_search(std::shared_ptr<const dataset<T>> dataset,
->UseRealTime();

if (metric_objective == Mode::kThroughput) { b->ThreadRange(threads[0], threads[1]); }
if (metric_objective == Mode::kThroughput) { b->ThreadRange(threads[0], threads[1]); }
Copy link
Contributor

Choose a reason for hiding this comment

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

I guess, this can be removed now?

Suggested change
if (metric_objective == Mode::kThroughput) { b->ThreadRange(threads[0], threads[1]); }

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CMake cpp improvement Improves an existing functionality non-breaking Introduces a non-breaking change Python
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants