diff --git a/c_api/IndexIVF_c_ex.cpp b/c_api/IndexIVF_c_ex.cpp index 39041a5265..989264a471 100644 --- a/c_api/IndexIVF_c_ex.cpp +++ b/c_api/IndexIVF_c_ex.cpp @@ -12,6 +12,7 @@ #include "macros_impl.h" using faiss::IndexIVF; +using faiss::SearchParameters; using faiss::SearchParametersIVF; int faiss_IndexIVF_set_direct_map(FaissIndexIVF* index, int direct_map_type) { @@ -39,12 +40,14 @@ int faiss_Search_closest_eligible_centroids( float* query, int k, float* centroid_distances, - idx_t* centroid_ids) { + idx_t* centroid_ids, + const FaissSearchParameters* params) { try { faiss::IndexIVF* index_ivf = reinterpret_cast(index); assert(index_ivf); - index_ivf->quantizer->search(n, query, k, centroid_distances, centroid_ids); + index_ivf->quantizer->search(n, query, k, centroid_distances, centroid_ids, + reinterpret_cast(params)); } CATCH_AND_HANDLE } @@ -66,4 +69,4 @@ int faiss_IndexIVF_search_preassigned_with_params( reinterpret_cast(params)); } CATCH_AND_HANDLE -} \ No newline at end of file +} diff --git a/c_api/IndexIVF_c_ex.h b/c_api/IndexIVF_c_ex.h index 1aa9378b66..e2effd3525 100644 --- a/c_api/IndexIVF_c_ex.h +++ b/c_api/IndexIVF_c_ex.h @@ -41,7 +41,8 @@ int faiss_Search_closest_eligible_centroids( float* query, int k, float* centroid_distances, - idx_t* centroid_ids + idx_t* centroid_ids, + const FaissSearchParameters* params ); /*