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.

Sign up for GitHub

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

Benchmark brute force knn #2063

Merged
merged 12 commits into from
Dec 20, 2023
+74 −49
Merged

Benchmark brute force knn #2063

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
49970f4
Benchmark brute force knn
benfred Dec 13, 2023
1fa690b
.
benfred Dec 13, 2023
aa98240
.
benfred Dec 13, 2023
19d7802
Merge branch 'branch-24.02' into benchmark_bfknn
benfred Dec 13, 2023
e1fada6
Adding yaml config files for brute force
cjnolet Dec 13, 2023
850d45b
fix build
benfred Dec 13, 2023
49a3ac0
fix for no build/search params
benfred Dec 13, 2023
6a7cbb7
fix faiss_gpu_flat/faiss_cpu_flat
benfred Dec 13, 2023
0656349
Merge branch 'branch-24.02' into benchmark_bfknn
cjnolet Dec 14, 2023
f71f657
add static assert
benfred Dec 15, 2023
fc4fdcd
Merge branch 'benchmark_bfknn' of github.com:benfred/raft into benchm…
benfred Dec 15, 2023
a8bb0aa
Merge branch 'branch-24.02' into benchmark_bfknn
benfred Dec 19, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
add static assert
  • Loading branch information
@benfred
benfred committed Dec 15, 2023
commit f71f6578dd39a1e9885177bd186832aa8f54cdf7
2 changes: 2 additions & 0 deletions cpp/bench/ann/src/raft/raft_wrapper.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,8 @@ template <typename T>
RaftGpu<T>::RaftGpu(Metric metric, int dim)
: ANN<T>(metric, dim), metric_type_(raft_temp::parse_metric_type(metric))
{
static_assert(std::is_same_v<T, float> || std::is_same_v<T, double>,
"raft bfknn only supports float/double");
RAFT_CUDA_TRY(cudaGetDevice(&device_));
}

Expand Down