From 10f0c259dc1c3beecdc1e610dca948a00c4ce8c9 Mon Sep 17 00:00:00 2001
From: psanal35 <pakize.sanal@amd.com>
Date: Wed, 11 Feb 2026 12:00:06 -0600
Subject: [PATCH 1/2] Update runner labels in Llama perf workflow

---
 .github/workflows/llama-perf.yml                | 12 ++++++------
 jax_rocm_plugin/build/rocm/targets_to_ignore.sh |  6 +-----
 2 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/llama-perf.yml b/.github/workflows/llama-perf.yml
index 11e42ce0bf..3cd5a8da4b 100644
--- a/.github/workflows/llama-perf.yml
+++ b/.github/workflows/llama-perf.yml
@@ -22,9 +22,9 @@ on:
         required: false
         type: choice
         options:
-          - linux-mi325-8
-          - linux-mi355-8
-        default: "linux-mi355-8"
+          - linux-jax-mi325-8
+          - linux-jax-mi355-8
+        default: "linux-jax-mi355-8"
       te-ref:
         required: false
         type: string
@@ -35,7 +35,7 @@ concurrency:
 
 jobs:
   build-te-wheels:
-    runs-on: linux-mi355-1
+    runs-on: linux-jax-mi355-1
     outputs:
       runner_label: ${{ steps.arch.outputs.label }}
       te_commit_sha: ${{ steps.tsha.outputs.hash }}
@@ -74,7 +74,7 @@ jobs:
       - name: Define ROCm arch & runner-label
         id: arch
         run: |
-          if [ "${{ inputs.runner-label }}" == "linux-mi325-8" ]; then
+          if [ "${{ inputs.runner-label }}" == "linux-jax-mi325-8" ]; then
             echo "ROCM_ARCH=gfx942" >> $GITHUB_ENV
             echo "label=MI325" >> $GITHUB_OUTPUT
           else
@@ -127,7 +127,7 @@ jobs:
           docker rm te-build-jax${{ matrix.jax-version }} || true
   run-llama-perf-model:
     needs: build-te-wheels
-    runs-on: ${{ inputs.runner-label || 'linux-mi355-8' }}
+    runs-on: ${{ inputs.runner-label || 'linux-jax-mi355-8' }}
     strategy:
       fail-fast: false
       max-parallel: 1
diff --git a/jax_rocm_plugin/build/rocm/targets_to_ignore.sh b/jax_rocm_plugin/build/rocm/targets_to_ignore.sh
index f91649b3a2..b9f41b020f 100755
--- a/jax_rocm_plugin/build/rocm/targets_to_ignore.sh
+++ b/jax_rocm_plugin/build/rocm/targets_to_ignore.sh
@@ -13,10 +13,6 @@ TARGETS_TO_IGNORE=(
     -@jax//tests:gpu_memory_flags_test_gpu
     -@jax//tests:lax_numpy_test_gpu
     -@jax//tests:lax_test_gpu
-    -@jax//tests:linalg_test_gpu
 )
 
-echo "$(
-    IFS=' '
-    echo "${TARGETS_TO_IGNORE[*]}"
-)"
+echo "${TARGETS_TO_IGNORE[*]}"

From 7b140cbe56095028f0cd5f4c3aca1cadca8affd6 Mon Sep 17 00:00:00 2001
From: Alexandros Theodoridis <atheodor@amd.com>
Date: Thu, 12 Feb 2026 14:58:41 +0000
Subject: [PATCH 2/2] Bump xla

---
 jax_rocm_plugin/third_party/xla/workspace.bzl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/jax_rocm_plugin/third_party/xla/workspace.bzl b/jax_rocm_plugin/third_party/xla/workspace.bzl
index 10645fe5b6..884ae163cc 100644
--- a/jax_rocm_plugin/third_party/xla/workspace.bzl
+++ b/jax_rocm_plugin/third_party/xla/workspace.bzl
@@ -16,8 +16,8 @@ load("//third_party:repo.bzl", "amd_http_archive")
 #    curl -L https://github.com/openxla/xla/archive/<git hash>.tar.gz | sha256sum
 #    and update XLA_SHA256 with the result.
 
-XLA_COMMIT = "24c5f10ae8fc24aefd20b43c501ade7f66fd0cfd"
-XLA_SHA256 = "f00db8761e86bcb51b52e64bc983717181050c8752c040e33ecb1429d861c30b"
+XLA_COMMIT = "a15e41b80f82e4160ba78851d80eaa90cb7e8435"
+XLA_SHA256 = "777f137337ac80ed72a77f82d1e610008a217f0daed384243fac676806919ffd"
 
 def repo():
     amd_http_archive(