From e9cee487d1ad109d4f916fb5a8fc2fc6682a909f Mon Sep 17 00:00:00 2001
From: Qian Sun <qian.sun2020@gmail.com>
Date: Mon, 18 Dec 2023 16:08:44 +0800
Subject: [PATCH] [GLUTEN-4080][VL] Add parameter --velox_repo & --velox_branch
 in builddeps-veloxbe.sh

---
 dev/builddeps-veloxbe.sh | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/dev/builddeps-veloxbe.sh b/dev/builddeps-veloxbe.sh
index dc3274119038..f714442e20e2 100755
--- a/dev/builddeps-veloxbe.sh
+++ b/dev/builddeps-veloxbe.sh
@@ -25,6 +25,8 @@ ENABLE_EP_CACHE=OFF
 ARROW_ENABLE_CUSTOM_CODEC=OFF
 ENABLE_VCPKG=OFF
 RUN_SETUP_SCRIPT=ON
+VELOX_REPO=https://github.com/oap-project/velox.git
+VELOX_BRANCH=update
 
 for arg in "$@"
 do
@@ -95,6 +97,14 @@ do
         RUN_SETUP_SCRIPT=("${arg#*=}")
         shift # Remove argument name from processing
         ;;
+	--velox_repo=*)
+ 	VELOX_REPO=("${arg#*=}")
+  	shift # Remove argument name from processing
+   	;;
+    	--velox_branch=*)
+        VELOX_BRANCH=("${arg#*=}")
+	shift # Remove argument name from processing
+ 	;;
 	      *)
         OTHER_ARGUMENTS+=("$1")
         shift # Remove generic argument from processing
@@ -110,7 +120,8 @@ fi
 
 ##install velox
 cd $GLUTEN_DIR/ep/build-velox/src
-./get_velox.sh --enable_hdfs=$ENABLE_HDFS --build_protobuf=$BUILD_PROTOBUF --enable_s3=$ENABLE_S3 --enable_gcs=$ENABLE_GCS --enable_abfs=$ENABLE_ABFS
+./get_velox.sh --enable_hdfs=$ENABLE_HDFS --build_protobuf=$BUILD_PROTOBUF --enable_s3=$ENABLE_S3 --enable_gcs=$ENABLE_GCS --enable_abfs=$ENABLE_ABFS \
+               --velox_repo=$VELOX_REPO --velox_branch=$VELOX_BRANCH
 ./build_velox.sh --run_setup_script=$RUN_SETUP_SCRIPT --enable_s3=$ENABLE_S3 --enable_gcs=$ENABLE_GCS --build_type=$BUILD_TYPE --enable_hdfs=$ENABLE_HDFS \
                  --enable_abfs=$ENABLE_ABFS --enable_ep_cache=$ENABLE_EP_CACHE --build_tests=$BUILD_TESTS --build_benchmarks=$BUILD_BENCHMARKS