From 58e591ae7c1d24b929f7a542fead68fdae803e3c Mon Sep 17 00:00:00 2001
From: quicksilver <zhifeng.zhang@zilliz.com>
Date: Wed, 20 Nov 2019 14:19:31 +0800
Subject: [PATCH] Update Jenkinsfile

---
 ci/jenkins/Jenkinsfile | 40 ++++++++++++++++++++--------------------
 ci/scripts/build.sh    | 10 ++++++++--
 2 files changed, 28 insertions(+), 22 deletions(-)

diff --git a/ci/jenkins/Jenkinsfile b/ci/jenkins/Jenkinsfile
index 8d3953b11..bf7a70d11 100644
--- a/ci/jenkins/Jenkinsfile
+++ b/ci/jenkins/Jenkinsfile
@@ -50,7 +50,7 @@ pipeline {
                     }
 
                     stages {
-                        stage("Run Build") {
+                        stage("Run GPU Version Build") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-build"
@@ -60,7 +60,7 @@ pipeline {
                             }
 
                             stages {
-                                stage('Build') {
+                                stage('GPU Version Build') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -69,7 +69,7 @@ pipeline {
                                         }
                                     }
                                 }
-                                stage('Code Coverage') {
+                                stage('GPU Version Code Coverage') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -78,7 +78,7 @@ pipeline {
                                         }
                                     }
                                 }
-                                stage('Upload Package') {
+                                stage('Upload GPU Version Package') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -90,7 +90,7 @@ pipeline {
                             }
                         }
 
-                        stage("Publish docker images") {
+                        stage("Publish GPU Version docker images") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-publish"
@@ -100,7 +100,7 @@ pipeline {
                             }
 
                             stages {
-                                stage('Publish') {
+                                stage('Publish GPU Version') {
                                     steps {
                                         container('publish-images'){
                                             script {
@@ -112,7 +112,7 @@ pipeline {
                             }
                         }
 
-                        stage("Deploy to Development") {
+                        stage("Deploy GPU Version to Development") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-dev-test"
@@ -122,7 +122,7 @@ pipeline {
                             }
 
                             stages {
-                                stage("Deploy to Dev") {
+                                stage("Deploy GPU Version to Dev") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
@@ -132,7 +132,7 @@ pipeline {
                                     }
                                 }
 
-                                stage("Dev Test") {
+                                stage("GPU Version Dev Test") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
@@ -147,7 +147,7 @@ pipeline {
                                     }
                                 }
 
-                                stage ("Cleanup Dev") {
+                                stage ("Cleanup GPU Version Dev") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
@@ -180,7 +180,7 @@ pipeline {
                     }
 
                     stages {
-                        stage("Run Build") {
+                        stage("Run CPU Version Build") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-build"
@@ -190,7 +190,7 @@ pipeline {
                             }
 
                             stages {
-                                stage('Build') {
+                                stage('Build CPU Version') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -199,7 +199,7 @@ pipeline {
                                         }
                                     }
                                 }
-                                stage('Code Coverage') {
+                                stage('CPU Version Code Coverage') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -208,7 +208,7 @@ pipeline {
                                         }
                                     }
                                 }
-                                stage('Upload Package') {
+                                stage('Upload CPU Version Package') {
                                     steps {
                                         container('milvus-build-env') {
                                             script {
@@ -220,7 +220,7 @@ pipeline {
                             }
                         }
 
-                        stage("Publish docker images") {
+                        stage("Publish CPU Version docker images") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-publish"
@@ -230,7 +230,7 @@ pipeline {
                             }
 
                             stages {
-                                stage('Publish') {
+                                stage('Publish CPU Version') {
                                     steps {
                                         container('publish-images'){
                                             script {
@@ -242,7 +242,7 @@ pipeline {
                             }
                         }
 
-                        stage("Deploy to Development") {
+                        stage("Deploy CPU Version to Development") {
                             agent {
                                 kubernetes {
                                     label "${BINRARY_VERSION}-dev-test"
@@ -252,7 +252,7 @@ pipeline {
                             }
 
                             stages {
-                                stage("Deploy to Dev") {
+                                stage("Deploy CPU Version to Dev") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
@@ -262,7 +262,7 @@ pipeline {
                                     }
                                 }
 
-                                stage("Dev Test") {
+                                stage("CPU Version Dev Test") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
@@ -277,7 +277,7 @@ pipeline {
                                     }
                                 }
 
-                                stage ("Cleanup Dev") {
+                                stage ("Cleanup CPU Version Dev") {
                                     steps {
                                         container('milvus-test-env') {
                                             script {
diff --git a/ci/scripts/build.sh b/ci/scripts/build.sh
index 27962ccb5..eb5205ad2 100755
--- a/ci/scripts/build.sh
+++ b/ci/scripts/build.sh
@@ -16,6 +16,7 @@ BUILD_TYPE="Debug"
 BUILD_UNITTEST="OFF"
 INSTALL_PREFIX="/opt/milvus"
 FAISS_ROOT=""
+CUSTOMIZATION="OFF" # default use origin faiss
 BUILD_COVERAGE="OFF"
 USE_JFROG_CACHE="OFF"
 RUN_CPPLINT="OFF"
@@ -23,7 +24,7 @@ GPU_VERSION="OFF"
 WITH_MKL="OFF"
 CUDA_COMPILER=/usr/local/cuda/bin/nvcc
 
-while getopts "o:t:b:f:gulcjmh" arg
+while getopts "o:t:b:f:gxulcjmh" arg
 do
         case $arg in
              o)
@@ -41,6 +42,9 @@ do
              g)
                 GPU_VERSION="ON";
                 ;;
+             x)
+                CUSTOMIZATION="ON";
+                ;;
              u)
                 echo "Build and run unittest cases" ;
                 BUILD_UNITTEST="ON";
@@ -66,6 +70,7 @@ parameter:
 -b: core code build directory
 -f: faiss root path
 -g: gpu version
+-x: milvus customization (default: OFF)
 -u: building unit test options(default: OFF)
 -l: run cpplint, clang-format and clang-tidy(default: OFF)
 -c: code coverage(default: OFF)
@@ -74,7 +79,7 @@ parameter:
 -h: help
 
 usage:
-./build.sh -o \${INSTALL_PREFIX} -t \${BUILD_TYPE} -b \${CORE_BUILD_DIR} -f \${FAISS_ROOT} [-u] [-l] [-c] [-j] [-m] [-h]
+./build.sh -o \${INSTALL_PREFIX} -t \${BUILD_TYPE} -b \${CORE_BUILD_DIR} -f \${FAISS_ROOT} [-g] [-x] [-u] [-l] [-c] [-j] [-m] [-h]
                 "
                 exit 0
                 ;;
@@ -96,6 +101,7 @@ CMAKE_CMD="cmake \
 -DCMAKE_BUILD_TYPE=${BUILD_TYPE} \
 -DCMAKE_CUDA_COMPILER=${CUDA_COMPILER} \
 -DMILVUS_GPU_VERSION=${GPU_VERSION} \
+-DCUSTOMIZATION=${CUSTOMIZATION} \
 -DBUILD_UNIT_TEST=${BUILD_UNITTEST} \
 -DBUILD_COVERAGE=${BUILD_COVERAGE} \
 -DUSE_JFROG_CACHE=${USE_JFROG_CACHE} \
-- 
GitLab