Setup latest dependency for new bufferization passes

tzunghanjuang · tzunghanjuang · commit 6af3eef5c727 · 2024-08-15T16:29:07.000-04:00
diff --git a/.dep-versions b/.dep-versions
@@ -1,5 +1,5 @@
 # Always update the version check in catalyst.__init__ when changing the JAX version.
-jax=0.4.28
+jax=0.4.31
 mhlo=89a891c986650c33df76885f5620e0a92150d90f
 llvm=3a8316216807d64a586b971f51695e23883331f7
 enzyme=v0.0.130
diff --git a/frontend/catalyst/__init__.py b/frontend/catalyst/__init__.py
@@ -23,7 +23,7 @@
 
 import jaxlib as _jaxlib
 
-_jaxlib_version = "0.4.28"
+_jaxlib_version = "0.4.31"
 if _jaxlib.__version__ != _jaxlib_version:
     import warnings
 
diff --git a/mlir/Makefile b/mlir/Makefile
@@ -118,7 +118,7 @@ enzyme:
 		-DCMAKE_CXX_VISIBILITY_PRESET=$(SYMBOL_VISIBILITY) \
 		-DCMAKE_POLICY_DEFAULT_CMP0116=NEW
 
-	cmake --build $(ENZYME_BUILD_DIR) --target EnzymeStatic-19
+	cmake --build $(ENZYME_BUILD_DIR) --target EnzymeStatic-20
 
 .PHONY: dialects
 dialects:
diff --git a/mlir/llvm-project b/mlir/llvm-project
@@ -1 +1 @@
-Subproject commit 3a8316216807d64a586b971f51695e23883331f7
+Subproject commit 51d4980a133db12888207698e39c469cb7055cac
diff --git a/mlir/mlir-hlo b/mlir/mlir-hlo
@@ -1 +1 @@
-Subproject commit 89a891c986650c33df76885f5620e0a92150d90f
+Subproject commit 1d151576543aebf134f76ae1ebf86da11623b912
