PR #23853: Enable the activation offloading test

Imported from GitHub PR jax-ml/jax#23853 This test ActivationOffloadingTest.test_remat_scan_layout_change_offloadable can be enabled after [XLA PR 17500](#17500) is in. This test is also a small reproducer for MaxText activation offloading --remat-policy=qkv_proj_offloaded. Copybara import of the project: -- adaf54a4bbe10ce05edcfeb29039c6948444c641 by Jane Liu <[email protected]>: enable the activation offloading test Merging this change closes #23853 PiperOrigin-RevId: 679365040
openxla · Sep 27, 2024 · f525272 · f525272
1 parent b57ddf1
commit f525272
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/python/xla_client.py b/xla/python/xla_client.py
@@ -50,7 +50,7 @@
 
 # Just an internal arbitrary increasing number to help with backward-compatible
 # changes. In JAX, reference this via jax._src.lib.xla_extension_version.
-_version = 288
+_version = 289
 
 # Version number for MLIR:Python components.
 mlir_api_version = 57