8000 CAGRA API update and allow async host refinement by mfoerste4 · Pull Request #131 · rapidsai/cuvs · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

CAGRA API update and allow async host refinement #131

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

Merged
merged 24 commits into from
May 30, 2024
Merged

CAGRA API update and allow async host refinement #131

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
4416b0b
enable asynchronous host-refinement for cagra index build
mfoerste4 May 17, 2024
946e106
move alloc out of OMP region
mfoerste4 May 17, 2024
7c6644e
added test for refinement -- disabled until detail::build API is exposed
mfoerste4 May 17, 2024
2c3f5e4
expose more cagra build parameters
mfoerste4 May 22, 2024
4d5ff0c
clarify code
mfoerste4 May 22, 2024
545032b
fix merge conflict
mfoerste4 May 22, 2024
d2c1dad
slightly reduce expected recall to catch refinement 1 results on all …
mfoerste4 May 24, 2024
9823831
cleanup API - use std::variant to switch algo of cagra graph build
mfoerste4 May 29, 2024
f310971
fix c-api nn_descent
mfoerste4 May 29, 2024
c1136c5
move documentation to API header
mfoerste4 May 29, 2024
bb87c97
resolve merge conflicts
mfoerste4 May 29, 2024
8506c6a
correct typo
mfoerste4 May 29, 2024
3a6e776
review suggestions
mfoerste4 May 29, 2024
42aa3eb
check valid nn_descent graph-degree / rename build_paramas to graph_b…
mfoerste4 May 29, 2024
68f490b
updating docs
mfoerste4 May 29, 2024
95f6064
Update CAGRA API (NN descent build by default)
tfeher May 30, 2024
3e22b13
Fix style
tfeher May 30, 2024
fd66402
remove cagra's build_host and build_device functions
tfeher May 30, 2024
8b1fbb8
remove debug printouts
tfeher May 30, 2024
e688453
Merge remote-tracking branch 'origin/branch-24.06' into cagra_ivfpq_i…
tfeher May 30, 2024
f5dff7d
fix style
tfeher May 30, 2024
71c779a
Merge branch 'branch-24.06' into cagra_ivfpq_index_async_refine
cjnolet May 30, 2024
2540dbf
Fix style
tfeher May 30, 2024
72d7b62
Fix CAGRA C API default algo params
tfeher May 30, 2024
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
2 changes: 2 additions & 0 deletions cpp/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,7 @@ add_library(
src/neighbors/cagra_serialize_float.cu
src/neighbors/cagra_serialize_int8.cu
src/neighbors/cagra_serialize_uint8.cu
src/neighbors/detail/cagra/cagra_build.cpp
src/neighbors/detail/cagra/q_search_multi_cta_float_uint32_dim128_t8_8pq_2subd_half.cu
src/neighbors/detail/cagra/q_search_multi_cta_float_uint32_dim128_t8_8pq_4subd_half.cu
src/neighbors/detail/cagra/q_search_multi_cta_float_uint32_dim256_t16_8pq_2subd_half.cu
Expand Down Expand Up @@ -425,6 +426,7 @@ add_library(
src/neighbors/ivf_pq/detail/ivf_pq_search_with_filter_float_int64_t.cu
src/neighbors/ivf_pq/detail/ivf_pq_search_with_filter_int8_t_int64_t.cu
src/neighbors/ivf_pq/detail/ivf_pq_search_with_filter_uint8_t_int64_t.cu
src/neighbors/nn_descent.cu
src/neighbors/nn_descent_float.cu
src/neighbors/nn_descent_int8.cu< B14A /td>
src/neighbors/nn_descent_uint8.cu
Expand Down
2 changes: 2 additions & 0 deletions cpp/include/cuvs/neighbors/cagra.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,8 @@ extern "C" {
*
*/
enum cuvsCagraGraphBuildAlgo {
/* Select build algorithm automatically */
AUTO_SELECT,
/* Use IVF-PQ to build all-neighbors knn graph */
IVF_PQ,
/* Experimental, use NN-Descent to build all-neighbors knn graph */
Expand Down
Loading
Loading
0