From 439456f8cddb952825c8736c6daec57e3918680e Mon Sep 17 00:00:00 2001
From: quicksilver <zhifeng.zhang@zilliz.com>
Date: Mon, 11 Nov 2019 19:48:29 +0800
Subject: [PATCH] add build with mkl option in build.sh

---
 ci/scripts/build.sh       | 10 ++++++++--
 ci/travis/travis_build.sh |  4 ++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/ci/scripts/build.sh b/ci/scripts/build.sh
index 118d6a4785..d077325fa8 100755
--- a/ci/scripts/build.sh
+++ b/ci/scripts/build.sh
@@ -19,9 +19,10 @@ BUILD_COVERAGE="OFF"
 USE_JFROG_CACHE="OFF"
 RUN_CPPLINT="OFF"
 CPU_VERSION="ON"
+WITH_MKL="OFF"
 CUDA_COMPILER=/usr/local/cuda/bin/nvcc
 
-while getopts "o:t:b:gulcjh" arg
+while getopts "o:t:b:gulcjmh" arg
 do
         case $arg in
              o)
@@ -49,6 +50,9 @@ do
              j)
                 USE_JFROG_CACHE="ON"
                 ;;
+             m)
+                WITH_MKL="ON"
+                ;;
              h) # help
                 echo "
 
@@ -61,10 +65,11 @@ parameter:
 -l: run cpplint, clang-format and clang-tidy(default: OFF)
 -c: code coverage(default: OFF)
 -j: use jfrog cache build directory(default: OFF)
+-m: build with MKL(default: OFF)
 -h: help
 
 usage:
-./build.sh -o \${INSTALL_PREFIX} -t \${BUILD_TYPE} -b \${CORE_BUILD_DIR} [-u] [-l] [-c] [-j] [-h]
+./build.sh -o \${INSTALL_PREFIX} -t \${BUILD_TYPE} -b \${CORE_BUILD_DIR} [-u] [-l] [-c] [-j] [-m] [-h]
                 "
                 exit 0
                 ;;
@@ -89,6 +94,7 @@ CMAKE_CMD="cmake \
 -DBUILD_UNIT_TEST=${BUILD_UNITTEST} \
 -DBUILD_COVERAGE=${BUILD_COVERAGE} \
 -DUSE_JFROG_CACHE=${USE_JFROG_CACHE} \
+-DBUILD_FAISS_WITH_MKL=${WITH_MKL} \
 -DARROW_SOURCE=AUTO \
 ${MILVUS_CORE_DIR}"
 echo ${CMAKE_CMD}
diff --git a/ci/travis/travis_build.sh b/ci/travis/travis_build.sh
index 3cde1d5a4d..3f49750b6d 100755
--- a/ci/travis/travis_build.sh
+++ b/ci/travis/travis_build.sh
@@ -18,7 +18,7 @@ done
 BUILD_COMMON_FLAGS="-t ${MILVUS_BUILD_TYPE} -o ${MILVUS_INSTALL_PREFIX} -b ${MILVUS_BUILD_DIR}"
 
 if [ $only_library_mode == "yes" ]; then
-  ${TRAVIS_BUILD_DIR}/ci/scripts/build.sh ${BUILD_COMMON_FLAGS}
+  ${TRAVIS_BUILD_DIR}/ci/scripts/build.sh ${BUILD_COMMON_FLAGS} -m
 else
-  ${TRAVIS_BUILD_DIR}/ci/scripts/build.sh ${BUILD_COMMON_FLAGS} -u -c
+  ${TRAVIS_BUILD_DIR}/ci/scripts/build.sh ${BUILD_COMMON_FLAGS} -m -u -c
 fi
\ No newline at end of file