mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
12f1d3c23c
Signed-off-by: ezeharia <ezeharia@gsitechnology.com> solve issue Support search over APU board #5202 support Tanimoto/Hamming Flat binary search over APU device . Search over APU can be done for FAISS_BIN_IDMAP engaine type only. Dependencies : build with FPGA (- a ) and APU (-w ) . build must contain FPGA flag . example : ./build.sh -a -w *** search is not possible without APU device conneted to a server .** Resolves: #5202
163 lines
3.2 KiB
Bash
Executable File
163 lines
3.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
BUILD_OUTPUT_DIR="cmake_build"
|
|
BUILD_TYPE="Debug"
|
|
BUILD_UNITTEST="OFF"
|
|
INSTALL_PREFIX=$(pwd)/milvus
|
|
MAKE_CLEAN="OFF"
|
|
BUILD_COVERAGE="OFF"
|
|
DB_PATH="/tmp/milvus"
|
|
PROFILING="OFF"
|
|
RUN_CPPLINT="OFF"
|
|
CUDA_COMPILER=/usr/local/cuda/bin/nvcc
|
|
GPU_VERSION="OFF" #defaults to CPU version
|
|
FPGA_VERSION="OFF"
|
|
APU_VERSION="OFF"
|
|
WITH_MKL="OFF"
|
|
WITH_PROMETHEUS="ON"
|
|
FIU_ENABLE="OFF"
|
|
BUILD_OPENBLAS="ON"
|
|
|
|
while getopts "p:d:t:f:ulrcgahzmeiw" arg; do
|
|
case $arg in
|
|
p)
|
|
INSTALL_PREFIX=$OPTARG
|
|
;;
|
|
d)
|
|
DB_PATH=$OPTARG
|
|
;;
|
|
t)
|
|
BUILD_TYPE=$OPTARG # BUILD_TYPE
|
|
;;
|
|
u)
|
|
echo "Build and run unittest cases"
|
|
BUILD_UNITTEST="ON"
|
|
;;
|
|
l)
|
|
RUN_CPPLINT="ON"
|
|
;;
|
|
r)
|
|
if [[ -d ${BUILD_OUTPUT_DIR} ]]; then
|
|
rm ./${BUILD_OUTPUT_DIR} -r
|
|
MAKE_CLEAN="ON"
|
|
fi
|
|
;;
|
|
c)
|
|
BUILD_COVERAGE="ON"
|
|
;;
|
|
z)
|
|
PROFILING="ON"
|
|
;;
|
|
g)
|
|
GPU_VERSION="ON"
|
|
;;
|
|
m)
|
|
WITH_MKL="ON"
|
|
;;
|
|
e)
|
|
WITH_PROMETHEUS="OFF"
|
|
;;
|
|
i)
|
|
FIU_ENABLE="ON"
|
|
;;
|
|
a)
|
|
FPGA_VERSION="ON"
|
|
;;
|
|
w)
|
|
APU_VERSION="ON"
|
|
;;
|
|
h) # help
|
|
echo "
|
|
|
|
parameter:
|
|
-p: install prefix(default: $(pwd)/milvus)
|
|
-d: db data path(default: /tmp/milvus)
|
|
-t: build type(default: Debug)
|
|
-u: building unit test options(default: OFF)
|
|
-l: run cpplint, clang-format and clang-tidy(default: OFF)
|
|
-r: remove previous build directory(default: OFF)
|
|
-c: code coverage(default: OFF)
|
|
-z: profiling(default: OFF)
|
|
-g: build GPU version(default: OFF)
|
|
-m: build with MKL(default: OFF)
|
|
-e: build without prometheus(default: OFF)
|
|
-i: build FIU_ENABLE(default: OFF)
|
|
-a: build FPGA(default: OFF)
|
|
-w: build APU(default: OFF )
|
|
-h: help
|
|
|
|
usage:
|
|
./build.sh -p \${INSTALL_PREFIX} -t \${BUILD_TYPE} [-u] [-l] [-r] [-c] [-z] [-g] [-a] [-m] [-e] [-h]
|
|
"
|
|
exit 0
|
|
;;
|
|
?)
|
|
echo "ERROR! unknown argument"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ ! -d ${BUILD_OUTPUT_DIR} ]]; then
|
|
mkdir ${BUILD_OUTPUT_DIR}
|
|
fi
|
|
|
|
cd ${BUILD_OUTPUT_DIR}
|
|
|
|
# remove make cache since build.sh -l use default variables
|
|
# force update the variables each time
|
|
make rebuild_cache >/dev/null 2>&1
|
|
|
|
CMAKE_CMD="cmake \
|
|
-DBUILD_UNIT_TEST=${BUILD_UNITTEST} \
|
|
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX}
|
|
-DCMAKE_BUILD_TYPE=${BUILD_TYPE} \
|
|
-DOpenBLAS_SOURCE=AUTO \
|
|
-DCMAKE_CUDA_COMPILER=${CUDA_COMPILER} \
|
|
-DBUILD_COVERAGE=${BUILD_COVERAGE} \
|
|
-DMILVUS_DB_PATH=${DB_PATH} \
|
|
-DENABLE_CPU_PROFILING=${PROFILING} \
|
|
-DMILVUS_GPU_VERSION=${GPU_VERSION} \
|
|
-DMILVUS_FPGA_VERSION=${FPGA_VERSION} \
|
|
-DMILVUS_APU_VERSION=${APU_VERSION} \
|
|
-DFAISS_WITH_MKL=${WITH_MKL} \
|
|
-DMILVUS_WITH_PROMETHEUS=${WITH_PROMETHEUS} \
|
|
-DMILVUS_WITH_FIU=${FIU_ENABLE} \
|
|
../"
|
|
echo ${CMAKE_CMD}
|
|
${CMAKE_CMD}
|
|
|
|
if [[ ${MAKE_CLEAN} == "ON" ]]; then
|
|
make clean
|
|
fi
|
|
|
|
if [[ ${RUN_CPPLINT} == "ON" ]]; then
|
|
# cpplint check
|
|
make lint
|
|
if [ $? -ne 0 ]; then
|
|
echo "ERROR! cpplint check failed"
|
|
exit 1
|
|
fi
|
|
echo "cpplint check passed!"
|
|
|
|
# clang-format check
|
|
make check-clang-format
|
|
if [ $? -ne 0 ]; then
|
|
echo "ERROR! clang-format check failed"
|
|
exit 1
|
|
fi
|
|
echo "clang-format check passed!"
|
|
|
|
# # clang-tidy check
|
|
# make check-clang-tidy
|
|
# if [ $? -ne 0 ]; then
|
|
# echo "ERROR! clang-tidy check failed"
|
|
# exit 1
|
|
# fi
|
|
# echo "clang-tidy check passed!"
|
|
else
|
|
|
|
# compile and build
|
|
make -j 8 install || exit 1
|
|
fi
|