Added 'j' option to use multiple cpus to build JNI library. (#2244)

Signed-off-by: Dooyong Kim <[email protected]> Co-authored-by: Dooyong Kim <[email protected]> (cherry picked from commit a029fa8)
opensearch-project · Nov 3, 2024 · ef3f656 · ef3f656
1 parent 1944783
commit ef3f656
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/scripts/build.sh b/scripts/build.sh
@@ -18,10 +18,11 @@ function usage() {
     echo -e "-p PLATFORM\t[Optional] Platform, ignored."
     echo -e "-a ARCHITECTURE\t[Optional] Build architecture, ignored."
     echo -e "-o OUTPUT\t[Optional] Output path, default is 'artifacts'."
+    echo -e "-j NPROC_COUNT\t[Optional] Number of CPUs to use when building JNI library. Default is 1."
     echo -e "-h help"
 }
 
-while getopts ":h:v:q:s:o:p:a:" arg; do
+while getopts ":h:v:q:s:o:p:a:j:" arg; do
     case $arg in
         h)
             usage
@@ -45,6 +46,9 @@ while getopts ":h:v:q:s:o:p:a:" arg; do
         a)
             ARCHITECTURE=$OPTARG
             ;;
+        j)
+            NPROC_COUNT=$OPTARG
+            ;;
         :)
             echo "Error: -${OPTARG} requires an argument"
             usage
@@ -118,7 +122,7 @@ fi
 # Build k-NN lib and plugin through gradle tasks
 cd $work_dir
 ./gradlew build --no-daemon --refresh-dependencies -x integTest -x test -Dopensearch.version=$VERSION -Dbuild.snapshot=$SNAPSHOT -Dbuild.version_qualifier=$QUALIFIER -Dbuild.lib.commit_patches=false
-./gradlew :buildJniLib -Davx512.enabled=false -Davx2.enabled=false -Dbuild.lib.commit_patches=false
+./gradlew :buildJniLib -Davx512.enabled=false -Davx2.enabled=false -Dbuild.lib.commit_patches=false -Dnproc.count=${NPROC_COUNT:-1}
 
 if [ "$PLATFORM" != "windows" ] && [ "$ARCHITECTURE" = "x64" ]; then
   echo "Building k-NN library after enabling AVX2"