diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index f9825a6..da37baa 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -29,7 +29,7 @@ jobs:
         with:
           repository: zml/xla
           token: ${{ secrets.OPENXLA_READ_ONLY }}
-          ref: 336122e2fb0e3d7dd93b5a4a30f7551d8e1a21b6
+          ref: 7018a84922e775445bcae422363703d2abe8d571
 
       - name: Setup Bazel
         run: |
diff --git a/third_party/openxla/xla.bzl b/third_party/openxla/xla.bzl
index ecb5940..ea3ba09 100644
--- a/third_party/openxla/xla.bzl
+++ b/third_party/openxla/xla.bzl
@@ -5,8 +5,8 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_file")
 # This version was chosen because it contains the gather/scatter broadcasting dims.
 # https://github.com/zml/xla/tree/gw/2024_09_19
 # Reference commit on xla@main: https://github.com/zml/xla/commit/d391119197eab771a84c1f8a59a7f50b7da4b43d
-OPENXLA_COMMIT = "177464c630b8acd2ba9795f81c0248bdaedfb3cf"
-OPENXLA_SHA256 = "48b33e15c4479c8c7421cfc5fa0c02f82af73e706c0ce7b383539cdeb55e3a41"
+OPENXLA_COMMIT = "7018a84922e775445bcae422363703d2abe8d571"
+OPENXLA_SHA256 = "3b01091dc340c4803b0ce9c57635ca37d0671a89bbaea0b788e45f364967fa1d"
 
 def _xla_impl(mctx):
     http_file(