From d3834a1d21122cf57ba315d9a6cb679d40569890 Mon Sep 17 00:00:00 2001
From: shivansh-ibm <shivansh.s1@ibm.com>
Date: Mon, 23 Mar 2026 03:48:27 -0700
Subject: [PATCH] update jaxlib to install requirred bazel version

---
 j/jaxlib/build_info.json   |  4 ++--
 j/jaxlib/jaxlib_ubi_9.3.sh | 38 +++++++++++++++++++++++---------------
 2 files changed, 25 insertions(+), 17 deletions(-)

diff --git a/j/jaxlib/build_info.json b/j/jaxlib/build_info.json
index 59f66c24de..a3be19de0e 100644
--- a/j/jaxlib/build_info.json
+++ b/j/jaxlib/build_info.json
@@ -1,8 +1,8 @@
 {
-  "maintainer": "stutiibm",
+  "maintainer": "shivansh-ibm",
   "package_name": "jaxlib",
   "github_url": "https://github.com/jax-ml/jax",
-  "version": "jaxlib-v0.4.7",
+  "version": "jaxlib-v0.4.23",
   "wheel_build" : true,
   "package_dir": "j/jaxlib",
   "default_branch": "main",
diff --git a/j/jaxlib/jaxlib_ubi_9.3.sh b/j/jaxlib/jaxlib_ubi_9.3.sh
index bd682d85c6..274ada3714 100644
--- a/j/jaxlib/jaxlib_ubi_9.3.sh
+++ b/j/jaxlib/jaxlib_ubi_9.3.sh
@@ -2,13 +2,13 @@
 # -----------------------------------------------------------------------------
 #
 # Package          : jaxlib
-# Version          : jaxlib-v0.4.7
+# Version          : jaxlib-v0.4.23
 # Source repo      : https://github.com/jax-ml/jax
 # Tested on        : UBI:9.3
 # Language         : Python
 # Ci-Check     : True
 # Script License   : Apache License, Version 2 or later
-# Maintainer       : Stuti Wali <Stuti.Wali@ibm.com>
+# Maintainer       : Shivansh Sharma <Shivansh.s1@ibm.com>
 #
 # Disclaimer       : This script has been tested in root mode on given
 # ==========         platform using the mentioned version of the package.
@@ -21,7 +21,7 @@
 
 # Variables
 PACKAGE_NAME=jax
-PACKAGE_VERSION=${1:-jaxlib-v0.4.7}
+PACKAGE_VERSION=${1:-jaxlib-v0.4.23}
 PACKAGE_URL=https://github.com/jax-ml/jax
 CURRENT_DIR=$pwd
 
@@ -39,26 +39,34 @@ yum install -y java-11-openjdk-devel  libtool xz  libevent-devel  clang java-11-
 export JAVA_HOME=/usr/lib/jvm/$(ls /usr/lib/jvm/ | grep -P '^(?=.*java-11)(?=.*ppc64le)')
 export PATH=$JAVA_HOME/bin:$PATH
 export LD_LIBRARY_PATH=/usr/lib64/:$LD_LIBRARY_PATH
- 
-# dnf groupinstall -y "Development Tools"
- 
-#installing bazel from source
+
+# Install Bazel based on .bazelversion
 echo "Installing bazel -------------------------------------------------------------"
+BAZEL_VERSION=""
+
+if [ -f ".bazelversion" ]; then
+    BAZEL_VERSION=$(cat .bazelversion | tr -d '[:space:]')
+    echo "Found .bazelversion: $BAZEL_VERSION"
+else
+    echo ".bazelversion not found, falling back to default"
+    BAZEL_VERSION="5.1.1"
+fi
+
+echo "Installing Bazel version: $BAZEL_VERSION"
 mkdir bazel
 cd bazel
-wget https://github.com/bazelbuild/bazel/releases/download/5.1.1/bazel-5.1.1-dist.zip
-unzip bazel-5.1.1-dist.zip
-echo "Installing bazel -------------------------------------------------------------"
+wget https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-dist.zip
+unzip bazel-${BAZEL_VERSION}-dist.zip
 env EXTRA_BAZEL_ARGS="--tool_java_runtime_version=local_jdk" bash ./compile.sh
 cp output/bazel /usr/local/bin
 export PATH=/usr/local/bin:$PATH
 bazel --version
-cd ..
- 
+cd $CURRENT_DIR
+
 echo "Installing dependencies via pip3-------------------------------------------------------------"
 pip3 install numpy==1.26.4 scipy wheel pytest
 pip3 install numpy==1.26.4 opt-einsum==3.3.0  ml-dtypes==0.5.0 absl-py 
- 
+
 # Clone the repository
 git clone $PACKAGE_URL
 cd $PACKAGE_NAME
@@ -78,9 +86,9 @@ http_archive(
 EOF
 )
 echo "$BORINGSSL_SUPPORT_CONTENT" > WORKSPACE-TEMP
-cat WORKSPACE >> WORKSPACE-TEMP 
+cat WORKSPACE >> WORKSPACE-TEMP
 rm -rf WORKSPACE && mv WORKSPACE-TEMP WORKSPACE
- 
+
 cd build
 #Install
 echo "Building package-------------------------------------------------------------"