mirrors/faiss
1
0
Fork 0
mirror of https://github.com/facebookresearch/faiss.git synced 2025-06-03 21:54:02 +08:00
Code Issues Projects Releases Wiki Activity

Pass store_dataset argument along to cuVS CAGRA (#4173)

Browse Source 
Summary:
This is required to enable lazy setting of a device copy of the training dataset to a cuVS CAGRA index.

Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4173

Reviewed By: mnorris11

Differential Revision: D69795662

Pulled By: gtwang01

fbshipit-source-id: 68cda198ed7983800b64d3e5fac1b77ff55ecd12
This commit is contained in:
Divye Gala 2025-02-20 19:30:30 -08:00 committed by Facebook GitHub Bot
parent d72d0cab6b
commit 6b652892ff
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
faiss/gpu/impl/CuvsCagra.cu
View File

@ -69,6 +69,7 @@ CuvsCagra::CuvsCagra(
index_params_.intermediate_graph_degree = intermediate_graph_degree;
index_params_.graph_degree = graph_degree;
index_params_.attach_dataset_on_build = store_dataset;
if (!ivf_pq_search_params_) {
ivf_pq_search_params_ =
@ -243,6 +244,7 @@ void CuvsCagra::search(
storage_, n_, dim_);
cuvs_index->update_dataset(raft_handle, dataset);
}
store_dataset_ = true;
}
auto queries_view = raft::make_device_matrix_view<const float, int64_t>(