move to correct file

blevesearch · Jan 11, 2024 · 50990d8 · 50990d8
1 parent ac3ca14
commit 50990d8
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 15 deletions.
diff --git a/c_api/IndexIVF_c.cpp b/c_api/IndexIVF_c.cpp
@@ -47,17 +47,6 @@ int faiss_SearchParametersIVF_new_with(
     CATCH_AND_HANDLE
 }
 
-int faiss_SearchParametersIVF_new_with_sel(
-        FaissSearchParametersIVF** p_sp,
-        FaissIDSelector* sel) {
-    try {
-        SearchParametersIVF* sp = new SearchParametersIVF;
-        sp->sel = reinterpret_cast<faiss::IDSelector*>(sel);
-        *p_sp = reinterpret_cast<FaissSearchParametersIVF*>(sp);
-    }
-    CATCH_AND_HANDLE
-}
-
 DEFINE_GETTER_PERMISSIVE(SearchParametersIVF, const FaissIDSelector*, sel)
 
 DEFINE_GETTER(SearchParametersIVF, size_t, nprobe)

diff --git a/c_api/IndexIVF_c.h b/c_api/IndexIVF_c.h
@@ -31,10 +31,6 @@ int faiss_SearchParametersIVF_new_with(
         size_t nprobe,
         size_t max_codes);
 
-int faiss_SearchParametersIVF_new_with_sel(
-        FaissSearchParametersIVF** p_sp,
-        FaissIDSelector* sel);
-
 FAISS_DECLARE_GETTER(SearchParametersIVF, const FaissIDSelector*, sel)
 FAISS_DECLARE_GETTER_SETTER(SearchParametersIVF, size_t, nprobe)
 FAISS_DECLARE_GETTER_SETTER(SearchParametersIVF, size_t, max_codes)

diff --git 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::SearchParametersIVF;
 
 int faiss_IndexIVF_set_direct_map(FaissIndexIVF* index, int direct_map_type) {
     try {
@@ -20,3 +21,14 @@ int faiss_IndexIVF_set_direct_map(FaissIndexIVF* index, int direct_map_type) {
     }
     CATCH_AND_HANDLE
 }
+
+int faiss_SearchParametersIVF_new_with_sel(
+        FaissSearchParametersIVF** p_sp,
+        FaissIDSelector* sel) {
+    try {
+        SearchParametersIVF* sp = new SearchParametersIVF;
+        sp->sel = reinterpret_cast<faiss::IDSelector*>(sel);
+        *p_sp = reinterpret_cast<FaissSearchParametersIVF*>(sp);
+    }
+    CATCH_AND_HANDLE
+}
diff --git a/c_api/IndexIVF_c_ex.h b/c_api/IndexIVF_c_ex.h
@@ -23,6 +23,10 @@ int faiss_IndexIVF_set_direct_map(
         FaissIndexIVF* index,
         int direct_map_type);
 
+int faiss_SearchParametersIVF_new_with_sel(
+        FaissSearchParametersIVF** p_sp,
+        FaissIDSelector* sel);
+
 #ifdef __cplusplus
 }
 #endif