From 05e20a9a5bc23d64aada066bb5aae809d1e525e6 Mon Sep 17 00:00:00 2001
From: Naveen Tatikonda <navtat@amazon.com>
Date: Fri, 2 Feb 2024 16:40:28 -0600
Subject: [PATCH] Update Build Script to build libraries with SIMD

Signed-off-by: Naveen Tatikonda <navtat@amazon.com>
---
 scripts/build.sh | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/scripts/build.sh b/scripts/build.sh
index 6a9adc256e..d937cefcac 100644
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -148,6 +148,20 @@ cd $distributions
 zip -ur $zipPath lib
 cd $work_dir
 
+if [ "$PLATFORM" != "windows" ]; then
+  echo "Building k-NN libraries after enabling SIMD"
+  ./gradlew :buildJniLib -Dsimd.enabled=true
+  mkdir $distributions/simd_lib
+  cp -v $ompPath $distributions/simd_lib
+  cp -v ./jni/release/${libPrefix}* $distributions/simd_lib
+  ls -l $distributions/simd_lib
+
+  # Add simd_lib directory to the k-NN plugin zip
+  cd $distributions
+  zip -ur $zipPath simd_lib
+  cd $work_dir
+fi
+
 echo "COPY ${distributions}/*.zip"
 mkdir -p $OUTPUT/plugins
 cp -v ${distributions}/*.zip $OUTPUT/plugins