-
Notifications
You must be signed in to change notification settings - Fork 128
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Skip rebuild from scratch after cmake is ran
By commiting the patch changes, this commit prevents the libraries to be built from scratch every time cmake is ran. This will save additional build time. In addition to this, this commit separates out the different functionality to make the build system more readable. We can continue to iterate on this in the future. Signed-off-by: John Mazanec <[email protected]>
- Loading branch information
1 parent
a2bb6cc
commit cbfcf38
Showing
5 changed files
with
120 additions
and
128 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,66 @@ | ||
| # | ||
| # Copyright OpenSearch Contributors | ||
| # SPDX-License-Identifier: Apache-2.0 | ||
| # | ||
|
|
||
| # Check if faiss exists | ||
| find_path(FAISS_REPO_DIR NAMES faiss PATHS ${CMAKE_CURRENT_SOURCE_DIR}/external/faiss NO_DEFAULT_PATH) | ||
|
|
||
| # If not, pull the updated submodule | ||
| if (NOT EXISTS ${FAISS_REPO_DIR}) | ||
| message(STATUS "Could not find faiss. Pulling updated submodule.") | ||
| execute_process(COMMAND git submodule update --init -- external/faiss WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}) | ||
| endif () | ||
|
|
||
| # Check if patch exist, this is to skip git apply during CI build. See CI.yml with ubuntu. | ||
| find_path(PATCH_FILE NAMES 0001-Custom-patch-to-support-multi-vector.patch 0002-Enable-precomp-table-to-be-shared-ivfpq.patch PATHS ${CMAKE_CURRENT_SOURCE_DIR}/patches/faiss NO_DEFAULT_PATH) | ||
|
|
||
| # If it exists, apply patches | ||
| if (EXISTS ${PATCH_FILE}) | ||
| message(STATUS "Applying custom patches.") | ||
| execute_process(COMMAND git am --3way ${CMAKE_CURRENT_SOURCE_DIR}/patches/faiss/0001-Custom-patch-to-support-multi-vector.patch WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/external/faiss ERROR_VARIABLE ERROR_MSG RESULT_VARIABLE RESULT_CODE) | ||
| execute_process(COMMAND git am --3way ${CMAKE_CURRENT_SOURCE_DIR}/patches/faiss/0002-Enable-precomp-table-to-be-shared-ivfpq.patch WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/external/faiss ERROR_VARIABLE ERROR_MSG RESULT_VARIABLE RESULT_CODE) | ||
| if(RESULT_CODE) | ||
| message(FATAL_ERROR "Failed to apply patch:\n${ERROR_MSG}") | ||
| endif() | ||
| endif() | ||
|
|
||
| if (${CMAKE_SYSTEM_NAME} STREQUAL Darwin) | ||
| if(CMAKE_C_COMPILER_ID MATCHES "Clang\$") | ||
| set(OpenMP_C_FLAGS "-Xpreprocessor -fopenmp") | ||
| set(OpenMP_C_LIB_NAMES "omp") | ||
| set(OpenMP_omp_LIBRARY /usr/local/opt/libomp/lib/libomp.dylib) | ||
| endif() | ||
|
|
||
| if(CMAKE_CXX_COMPILER_ID MATCHES "Clang\$") | ||
| set(OpenMP_CXX_FLAGS "-Xpreprocessor -fopenmp -I/usr/local/opt/libomp/include") | ||
| set(OpenMP_CXX_LIB_NAMES "omp") | ||
| set(OpenMP_omp_LIBRARY /usr/local/opt/libomp/lib/libomp.dylib) | ||
| endif() | ||
| endif() | ||
|
|
||
| find_package(ZLIB REQUIRED) | ||
| find_package(BLAS REQUIRED) | ||
| enable_language(Fortran) | ||
| find_package(LAPACK REQUIRED) | ||
|
|
||
| # Set relevant properties | ||
| set(BUILD_TESTING OFF) # Avoid building faiss tests | ||
| set(BLA_STATIC ON) # Statically link BLAS | ||
| set(FAISS_ENABLE_GPU OFF) | ||
| set(FAISS_ENABLE_PYTHON OFF) | ||
|
|
||
| if(NOT DEFINED SIMD_ENABLED) | ||
| set(SIMD_ENABLED true) # set default value as true if the argument is not set | ||
| endif() | ||
|
|
||
| if(${CMAKE_SYSTEM_NAME} STREQUAL Windows OR ${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64" OR ${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm64" OR NOT ${SIMD_ENABLED}) | ||
| set(FAISS_OPT_LEVEL generic) # Keep optimization level as generic on Windows OS as it is not supported due to MINGW64 compiler issue. Also, on aarch64 avx2 is not supported. | ||
| set(TARGET_LINK_FAISS_LIB faiss) | ||
| else() | ||
| set(FAISS_OPT_LEVEL avx2) # Keep optimization level as avx2 to improve performance on Linux and Mac. | ||
| set(TARGET_LINK_FAISS_LIB faiss_avx2) | ||
| string(PREPEND LIB_EXT "_avx2") # Prepend "_avx2" to lib extension to create the library as "libopensearchknn_faiss_avx2.so" on linux and "libopensearchknn_faiss_avx2.jnilib" on mac | ||
| endif() | ||
|
|
||
| add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/external/faiss EXCLUDE_FROM_ALL) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,28 @@ | ||
| # | ||
| # Copyright OpenSearch Contributors | ||
| # SPDX-License-Identifier: Apache-2.0 | ||
| # | ||
|
|
||
| # Check if nmslib exists | ||
| find_path(NMS_REPO_DIR NAMES similarity_search PATHS ${CMAKE_CURRENT_SOURCE_DIR}/external/nmslib NO_DEFAULT_PATH) | ||
|
|
||
| # If not, pull the updated submodule | ||
| if (NOT EXISTS ${NMS_REPO_DIR}) | ||
| message(STATUS "Could not find nmslib. Pulling updated submodule.") | ||
| execute_process(COMMAND git submodule update --init -- external/nmslib WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}) | ||
| endif () | ||
|
|
||
| # Check if patch exist, this is to skip git apply during CI build. See CI.yml with ubuntu. | ||
| find_path(PATCH_FILE NAMES 0001-Initialize-maxlevel-during-add-from-enterpoint-level.patch PATHS ${CMAKE_CURRENT_SOURCE_DIR}/patches/nmslib NO_DEFAULT_PATH) | ||
|
|
||
| # If it exists, apply patches | ||
| if (EXISTS ${PATCH_FILE}) | ||
| message(STATUS "Applying custom patches.") | ||
| execute_process(COMMAND git am --3way ${CMAKE_CURRENT_SOURCE_DIR}/patches/nmslib/0001-Initialize-maxlevel-during-add-from-enterpoint-level.patch WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/external/nmslib ERROR_VARIABLE ERROR_MSG RESULT_VARIABLE RESULT_CODE) | ||
|
|
||
| if(RESULT_CODE) | ||
| message(FATAL_ERROR "Failed to apply patch:\n${ERROR_MSG}") | ||
| endif() | ||
| endif() | ||
|
|
||
| add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/external/nmslib/similarity_search) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,17 @@ | ||
| # | ||
| # Copyright OpenSearch Contributors | ||
| # SPDX-License-Identifier: Apache-2.0 | ||
| # | ||
|
|
||
| macro(opensearch_set_common_properties TARGET) | ||
| message(${TARGET}) | ||
| set_target_properties(${TARGET} PROPERTIES SUFFIX ${LIB_EXT}) | ||
| set_target_properties(${TARGET} PROPERTIES POSITION_INDEPENDENT_CODE ON) | ||
|
|
||
| if (NOT "${WIN32}" STREQUAL "") | ||
| # Use RUNTIME_OUTPUT_DIRECTORY, to build the target library (opensearchknn_faiss) in the specified directory at runtime. | ||
| set_target_properties(${TARGET} PROPERTIES RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/release) | ||
| else() | ||
| set_target_properties(${TARGET} PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/release) | ||
| endif() | ||
| endmacro() |