FindFAISS.cmake 1.3 KB
Newer Older
Z
Zhiru Zhu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
set(FAISS_STATIC_LIB_NAME ${CMAKE_STATIC_LIBRARY_PREFIX}faiss${CMAKE_STATIC_LIBRARY_SUFFIX})

# First, find via if specified FAISS_ROOT
if (FAISS_ROOT)
    find_library(FAISS_STATIC_LIB
            NAMES ${FAISS_STATIC_LIB_NAME}
            PATHS ${FAISS_ROOT}
            PATH_SUFFIXES "lib"
            NO_DEFAULT_PATH
            )
    find_path(FAISS_INCLUDE_DIR
            NAMES "faiss/Index.h"
            PATHS ${FAISS_ROOT}
            NO_DEFAULT_PATH
            PATH_SUFFIXES "include"
            )
endif ()

find_package_handle_standard_args(FAISS REQUIRED_VARS FAISS_STATIC_LIB FAISS_INCLUDE_DIR)

if (FAISS_FOUND)
    if (NOT TARGET faiss)
        add_library(faiss STATIC IMPORTED)

        set_target_properties(
                faiss
                PROPERTIES
                IMPORTED_LOCATION "${FAISS_STATIC_LIB}"
                INTERFACE_INCLUDE_DIRECTORIES "${FAISS_INCLUDE_DIR}"
        )

        if (FAISS_WITH_MKL)
            set_target_properties(
                    faiss
                    PROPERTIES
                    INTERFACE_LINK_LIBRARIES "${MKL_LIBS}")
        else ()
            set_target_properties(
                    faiss
                    PROPERTIES
                    INTERFACE_LINK_LIBRARIES "openblas;lapack")
        endif ()
    endif ()
endif ()