2020-08-25 16:56:01 -07:00
|
|
|
#!/bin/sh
|
|
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
|
|
#
|
|
|
|
# This source code is licensed under the MIT license found in the
|
|
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
|
|
|
|
set -e
|
|
|
|
|
|
|
|
|
2021-12-07 13:11:15 -08:00
|
|
|
# Build libfaiss.so/libfaiss_avx2.so.
|
2021-03-26 14:27:30 -07:00
|
|
|
cmake -B _build \
|
2020-08-25 16:56:01 -07:00
|
|
|
-DBUILD_SHARED_LIBS=ON \
|
|
|
|
-DBUILD_TESTING=OFF \
|
|
|
|
-DFAISS_OPT_LEVEL=avx2 \
|
|
|
|
-DFAISS_ENABLE_GPU=ON \
|
2020-09-23 10:16:56 -07:00
|
|
|
-DCMAKE_CUDA_ARCHITECTURES="${CUDA_ARCHS}" \
|
2020-08-25 16:56:01 -07:00
|
|
|
-DFAISS_ENABLE_PYTHON=OFF \
|
|
|
|
-DBLA_VENDOR=Intel10_64lp \
|
2021-12-07 13:11:15 -08:00
|
|
|
-DCMAKE_INSTALL_LIBDIR=lib \
|
2020-08-25 16:56:01 -07:00
|
|
|
-DCMAKE_BUILD_TYPE=Release .
|
|
|
|
|
2023-01-03 13:22:30 -08:00
|
|
|
make -C _build -j$(nproc) faiss faiss_avx2
|
2020-08-25 16:56:01 -07:00
|
|
|
|
2021-03-26 14:27:30 -07:00
|
|
|
cmake --install _build --prefix $PREFIX
|
|
|
|
cmake --install _build --prefix _libfaiss_stage/
|