From e109b0058df0599d869fac3e33279a94e3287cdc Mon Sep 17 00:00:00 2001 From: shivansh-ibm Date: Mon, 23 Mar 2026 03:48:27 -0700 Subject: [PATCH 1/2] update jaxlib to install requirred bazel version --- j/jaxlib/build_info.json | 4 ++-- j/jaxlib/jaxlib_ubi_9.3.sh | 39 +++++++++++++++++++++++--------------- 2 files changed, 26 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..cf7fa6ede1 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 +# Maintainer : Shivansh Sharma # # 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,35 @@ 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 +87,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-------------------------------------------------------------" From 363de5271ab6603ee766c671a35b35767c9c61b3 Mon Sep 17 00:00:00 2001 From: shivansh-ibm Date: Mon, 30 Mar 2026 12:34:21 +0000 Subject: [PATCH 2/2] update build script --- j/jaxlib/jaxlib_ubi_9.3.sh | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/j/jaxlib/jaxlib_ubi_9.3.sh b/j/jaxlib/jaxlib_ubi_9.3.sh index cf7fa6ede1..a6a6432b11 100644 --- a/j/jaxlib/jaxlib_ubi_9.3.sh +++ b/j/jaxlib/jaxlib_ubi_9.3.sh @@ -23,7 +23,7 @@ PACKAGE_NAME=jax PACKAGE_VERSION=${1:-jaxlib-v0.4.23} PACKAGE_URL=https://github.com/jax-ml/jax -CURRENT_DIR=$pwd +CURRENT_DIR=$(pwd) # Install dependencies echo "Installing dependencies -------------------------------------------------------------" @@ -40,8 +40,11 @@ export JAVA_HOME=/usr/lib/jvm/$(ls /usr/lib/jvm/ | grep -P '^(?=.*java-11)(?=.*p export PATH=$JAVA_HOME/bin:$PATH export LD_LIBRARY_PATH=/usr/lib64/:$LD_LIBRARY_PATH -# Install Bazel based on .bazelversion -echo "Installing bazel -------------------------------------------------------------" +# Clone the repository +git clone $PACKAGE_URL +cd $PACKAGE_NAME +git checkout $PACKAGE_VERSION + BAZEL_VERSION="" if [ -f ".bazelversion" ]; then @@ -53,7 +56,7 @@ else fi echo "Installing Bazel version: $BAZEL_VERSION" - +cd $CURRENT_DIR mkdir bazel cd bazel wget https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-dist.zip @@ -62,18 +65,13 @@ 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 $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 -git checkout $PACKAGE_VERSION +pip3 install numpy==1.26.4 scipy wheel pytest build +pip3 install numpy==1.26.4 opt-einsum==3.3.0 ml-dtypes==0.5.0 absl-py #Add boringssl to build for jaxlib +cd $CURRENT_DIR/$PACKAGE_NAME BORINGSSL_SUPPORT_CONTENT=$(cat << EOF load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") http_archive(