Bump faiss lib commit to 32f0e8cf92cd2275b60364517bb1cce67aa29a55

CHANGELOG.md

@@ -19,4 +19,5 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 ### Infrastructure
 ### Documentation
 ### Maintenance
+* Bump faiss lib commit to 32f0e8cf92cd2275b60364517bb1cce67aa29a55 [#1443](https://github.com/opensearch-project/k-NN/pull/1443)
 ### Refactoring

jni/CMakeLists.txt

@@ -181,17 +181,15 @@ if (${CONFIG_FAISS} STREQUAL ON OR ${CONFIG_ALL} STREQUAL ON OR ${CONFIG_TEST} S
         ${TARGET_LIB_FAISS} SHARED
         ${CMAKE_CURRENT_SOURCE_DIR}/src/org_opensearch_knn_jni_FaissService.cpp
         ${CMAKE_CURRENT_SOURCE_DIR}/src/faiss_wrapper.cpp
-        ${CMAKE_CURRENT_SOURCE_DIR}/src/knn_extension/faiss/utils/BitSet.cpp
-        ${CMAKE_CURRENT_SOURCE_DIR}/src/knn_extension/faiss/MultiVectorResultCollector.cpp
-        ${CMAKE_CURRENT_SOURCE_DIR}/src/knn_extension/faiss/MultiVectorResultCollectorFactory.cpp)
+        ${CMAKE_CURRENT_SOURCE_DIR}/src/faiss_util.cpp
+    )
     target_link_libraries(${TARGET_LIB_FAISS} ${TARGET_LINK_FAISS_LIB} ${TARGET_LIB_COMMON} OpenMP::OpenMP_CXX)
     target_include_directories(${TARGET_LIB_FAISS} PRIVATE
         ${CMAKE_CURRENT_SOURCE_DIR}/include
-        ${CMAKE_CURRENT_SOURCE_DIR}/include/knn_extension/faiss
-        ${CMAKE_CURRENT_SOURCE_DIR}/include/knn_extension/faiss/utils
         $ENV{JAVA_HOME}/include
         $ENV{JAVA_HOME}/include/${JVM_OS_TYPE}
-        ${CMAKE_CURRENT_SOURCE_DIR}/external/faiss)
+        ${CMAKE_CURRENT_SOURCE_DIR}/external/faiss
+    )
     set_target_properties(${TARGET_LIB_FAISS} PROPERTIES SUFFIX ${LIB_EXT})
     set_target_properties(${TARGET_LIB_FAISS} PROPERTIES POSITION_INDEPENDENT_CODE ON)
 
@@ -228,12 +226,9 @@ if ("${WIN32}" STREQUAL "")
         add_executable(
                 jni_test
                 tests/faiss_wrapper_test.cpp
+                tests/faiss_util_test.cpp
                 tests/nmslib_wrapper_test.cpp
                 tests/test_util.cpp
-                tests/knn_extension/faiss/utils/BitSetTest.cpp
-                tests/knn_extension/faiss/utils/HeapTest.cpp
-                tests/knn_extension/faiss/MultiVectorResultCollectorTest.cpp
-                tests/knn_extension/faiss/MultiVectorResultCollectorFactoryTest.cpp
                 )
 
         target_link_libraries(

jni/include/faiss_util.h

@@ -0,0 +1,25 @@
+// SPDX-License-Identifier: Apache-2.0
+//
+// The OpenSearch Contributors require contributions made to
+// this file be licensed under the Apache-2.0 license or a
+// compatible open source license.
+//
+// Modifications Copyright OpenSearch Contributors. See
+// GitHub history for details.
+
+/**
+ * This file contains util methods which are free of JNI to be used in faiss_wrapper.cpp
+ */
+
+#ifndef OPENSEARCH_KNN_FAISS_UTIL_H
+#define OPENSEARCH_KNN_FAISS_UTIL_H
+
+#include "faiss/impl/IDGrouper.h"
+#include <memory>
+
+namespace faiss_util {
+    std::unique_ptr<faiss::IDGrouperBitmap> buildIDGrouperBitmap(int *parentIdsArray,  int parentIdsLength, std::vector<uint64_t>* bitmap);
+};
+
+
+#endif //OPENSEARCH_KNN_FAISS_UTIL_H