Move XLA pin to b9ece5154181608ff1acea9b02a1ee7a526af16b

pytorch · Mar 21, 2024 · ac0e5eb · ac0e5eb
1 parent de42834
commit ac0e5eb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 44 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -49,11 +49,10 @@ http_archive(
         "//openxla_patches:cache_urls.diff",
         "//openxla_patches:gpu_race_condition.diff",
         "//openxla_patches:f16_abi_clang.diff",
-        "//openxla_patches:unbounded_dynamism.diff",
     ],
-    strip_prefix = "xla-25c8a6781af6be51d3bc43a0953b07803ab761ea",
+    strip_prefix = "xla-b9ece5154181608ff1acea9b02a1ee7a526af16b",
     urls = [
-        "https://github.com/openxla/xla/archive/25c8a6781af6be51d3bc43a0953b07803ab761ea.tar.gz",
+        "https://github.com/openxla/xla/archive/b9ece5154181608ff1acea9b02a1ee7a526af16b.tar.gz",
     ],
 )
 

diff --git a/openxla_patches/unbounded_dynamism.diff b/openxla_patches/unbounded_dynamism.diff
diff --git a/setup.py b/setup.py
@@ -64,7 +64,7 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240320'
+_date = '20240322'
 _libtpu_version = f'0.1.dev{_date}'
 _libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
 _jax_version = f'0.4.26.dev{_date}'