mirror of https://github.com/milvus-io/milvus.git
125 lines
2.9 KiB
Bash
Executable File
125 lines
2.9 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
BUILD_TYPE="Debug"
|
|
BUILD_UNITTEST="OFF"
|
|
INSTALL_PREFIX=$(pwd)/milvus
|
|
MAKE_CLEAN="OFF"
|
|
BUILD_COVERAGE="OFF"
|
|
DB_PATH="/opt/milvus"
|
|
PROFILING="OFF"
|
|
BUILD_FAISS_WITH_MKL="OFF"
|
|
USE_JFROG_CACHE="OFF"
|
|
KNOWHERE_BUILD_DIR="`pwd`/src/core/cmake_build"
|
|
KNOWHERE_OPTIONS="-t ${BUILD_TYPE}"
|
|
|
|
while getopts "p:d:t:k:uhrcgmj" arg
|
|
do
|
|
case $arg in
|
|
t)
|
|
BUILD_TYPE=$OPTARG # BUILD_TYPE
|
|
KNOWHERE_OPTIONS="-t ${BUILD_TYPE}"
|
|
;;
|
|
u)
|
|
echo "Build and run unittest cases" ;
|
|
BUILD_UNITTEST="ON";
|
|
;;
|
|
p)
|
|
INSTALL_PREFIX=$OPTARG
|
|
;;
|
|
d)
|
|
DB_PATH=$OPTARG
|
|
;;
|
|
r)
|
|
if [[ -d cmake_build ]]; then
|
|
rm ./cmake_build -r
|
|
MAKE_CLEAN="ON"
|
|
fi
|
|
;;
|
|
c)
|
|
BUILD_COVERAGE="ON"
|
|
;;
|
|
g)
|
|
PROFILING="ON"
|
|
;;
|
|
k)
|
|
KNOWHERE_BUILD_DIR=$OPTARG
|
|
;;
|
|
m)
|
|
BUILD_FAISS_WITH_MKL="ON"
|
|
;;
|
|
j)
|
|
USE_JFROG_CACHE="ON"
|
|
KNOWHERE_OPTIONS="${KNOWHERE_OPTIONS} -j"
|
|
;;
|
|
h) # help
|
|
echo "
|
|
|
|
parameter:
|
|
-t: build type(default: Debug)
|
|
-u: building unit test options(default: OFF)
|
|
-p: install prefix(default: $(pwd)/milvus)
|
|
-d: db path(default: /opt/milvus)
|
|
-r: remove previous build directory(default: OFF)
|
|
-c: code coverage(default: OFF)
|
|
-g: profiling(default: OFF)
|
|
-k: specify knowhere header/binary path
|
|
-m: build faiss with MKL(default: OFF)
|
|
-j: use jfrog cache build directory
|
|
|
|
usage:
|
|
./build.sh -t \${BUILD_TYPE} [-u] [-h] [-g] [-r] [-c] [-k] [-m] [-j]
|
|
"
|
|
exit 0
|
|
;;
|
|
?)
|
|
echo "unknown argument"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ ! -d cmake_build ]]; then
|
|
mkdir cmake_build
|
|
MAKE_CLEAN="ON"
|
|
fi
|
|
|
|
pushd `pwd`/src/core
|
|
./build.sh ${KNOWHERE_OPTIONS}
|
|
popd
|
|
|
|
cd cmake_build
|
|
git
|
|
CUDA_COMPILER=/usr/local/cuda/bin/nvcc
|
|
|
|
if [[ ${MAKE_CLEAN} == "ON" ]]; then
|
|
CMAKE_CMD="cmake -DBUILD_UNIT_TEST=${BUILD_UNITTEST} \
|
|
-DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX}
|
|
-DCMAKE_BUILD_TYPE=${BUILD_TYPE} \
|
|
-DCMAKE_CUDA_COMPILER=${CUDA_COMPILER} \
|
|
-DBUILD_COVERAGE=${BUILD_COVERAGE} \
|
|
-DMILVUS_DB_PATH=${DB_PATH} \
|
|
-DMILVUS_ENABLE_PROFILING=${PROFILING} \
|
|
-DBUILD_FAISS_WITH_MKL=${BUILD_FAISS_WITH_MKL} \
|
|
-DKNOWHERE_BUILD_DIR=${KNOWHERE_BUILD_DIR} \
|
|
-DUSE_JFROG_CACHE=${USE_JFROG_CACHE} \
|
|
../"
|
|
echo ${CMAKE_CMD}
|
|
|
|
${CMAKE_CMD}
|
|
make clean
|
|
fi
|
|
|
|
make -j 4 || exit 1
|
|
|
|
if [[ ${BUILD_TYPE} != "Debug" ]]; then
|
|
strip src/milvus_server
|
|
fi
|
|
|
|
make install || exit 1
|
|
|
|
if [[ ${BUILD_COVERAGE} == "ON" ]]; then
|
|
cd -
|
|
bash `pwd`/coverage.sh
|
|
cd -
|
|
fi
|