diff --git a/.github/workflows/IDC_1100_Public_CI.yml b/.github/workflows/IDC_1100_Public_CI.yml
index b3096a8..de39e7d 100644
--- a/.github/workflows/IDC_1100_Public_CI.yml
+++ b/.github/workflows/IDC_1100_Public_CI.yml
@@ -25,13 +25,14 @@ jobs:
         run: |
           set +e
           CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
-          source /opt/miniconda/bin/activate ${CONDA_ENV}
+          source /opt/conda/bin/activate ${CONDA_ENV}
           if [ $? -ne 0 ]; then
               echo 'conda env does not exist'
               conda create -n ${CONDA_ENV} python=3.10 -y
               conda activate ${CONDA_ENV}
           fi
-          pip install jax==0.4.20 jaxlib==0.4.20
+          pip install -r ./test/requirements.txt
+          #pip install jax==0.4.24 jaxlib==0.4.24
           conda install libstdcxx-ng==12.2.0 -c conda-forge
           pip install absl-py
           pip list | grep numpy
@@ -40,7 +41,7 @@ jobs:
         run: |
           set +e
           CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
-          source /opt/miniconda/bin/activate ${CONDA_ENV}
+          source /opt/conda/bin/activate ${CONDA_ENV}
           which bazel
           if [ $? -eq 1 ]; then
             wget https://github.com/bazelbuild/bazel/releases/download/5.3.0/bazel-5.3.0-installer-linux-x86_64.sh
@@ -70,7 +71,7 @@ jobs:
           echo $build_result
           if [ "$build_result" = "0" ];then
               echo "Build successful"
-              pip install *.whl
+               pip install --force-reinstall *.whl
           else
               echo "Build Failed"
               exit 1
@@ -79,13 +80,14 @@ jobs:
         run: |
           set +e
           CONDA_ENV=openxla_preCI_${{ github.event.pull_request.number }}_${{ runner.name }}
-          source /opt/miniconda/bin/activate ${CONDA_ENV}
+          source /opt/conda/bin/activate ${CONDA_ENV}
           basekit_path=/home/sdp
           source ${basekit_path}/intel/oneapi/compiler/latest/env/vars.sh
           source ${basekit_path}/intel/oneapi/mkl/latest/env/vars.sh
-          export ZE_AFFINITY_MASK=0,1
+          export ZE_AFFINITY_MASK=5,6
+          ut_branch=$(cat ./test/BRANCH_NAME)
           git clone https://github.com/google/jax.git
-          cd jax && git checkout jaxlib-v0.4.20
+          cd jax && git checkout $ut_branch
           patch -p1 -i ../test/jax_ut.patch
           pip install hypothesis
           pip install -e .
@@ -170,10 +172,10 @@ jobs:
         uses: actions/upload-artifact@v3
         with:
           name: XLA-CI-Data
-          path: /home/sdp/actions-runner/workspace_openxla/intel-extension-for-openxla/intel-extension-for-openxla/jax/logs
+          path: /home/sdp/actions-runner/_work/intel-extension-for-openxla/intel-extension-for-openxla/jax/logs
       - name: Test Results Check
         run: |
-          cd /home/sdp/actions-runner/workspace_openxla/intel-extension-for-openxla/intel-extension-for-openxla/jax
+          cd /home/sdp/actions-runner/_work/intel-extension-for-openxla/intel-extension-for-openxla/jax
           #UT results check
           if [ -f "./logs/ut/err.log" ]; then
             echo "exist fail cases"