From 725087e13f3438f743ed7e03a14c9436edcd877a Mon Sep 17 00:00:00 2001
From: jax authors <google-ml-automation@google.com>
Date: Tue, 18 Feb 2025 10:28:21 -0800
Subject: [PATCH] Integrate LLVM at llvm/llvm-project@9d24f9437944

Updates LLVM usage to match
[9d24f9437944](https://github.com/llvm/llvm-project/commit/9d24f9437944)

PiperOrigin-RevId: 728265165
---
 jaxlib/mosaic/gpu/custom_call.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/jaxlib/mosaic/gpu/custom_call.cc b/jaxlib/mosaic/gpu/custom_call.cc
index 5a3facbb2..361c839b6 100644
--- a/jaxlib/mosaic/gpu/custom_call.cc
+++ b/jaxlib/mosaic/gpu/custom_call.cc
@@ -130,15 +130,15 @@ mlir::FailureOr<mlir::OpPassManager> GetPassPipeline(
     mlir::registerStripDebugInfo();
     mlir::registerConvertNVGPUToNVVMPass();
     mlir::registerConvertVectorToSCF();
-    mlir::registerSCFToControlFlow();
+    mlir::registerSCFToControlFlowPass();
     mlir::registerConvertNVVMToLLVMPass();
     mlir::registerArithToLLVMConversionPass();
     mlir::registerConvertIndexToLLVMPass();
     mlir::registerConvertGpuOpsToNVVMOps();
     mlir::registerConvertMathToLLVMPass();
     mlir::registerConvertFuncToLLVMPass();
-    mlir::registerConvertAffineToStandard();
-    mlir::registerReconcileUnrealizedCasts();
+    mlir::registerLowerAffinePass();
+    mlir::registerReconcileUnrealizedCastsPass();
     // TODO(apaszke): Only register the passes we actually use.
     mlir::memref::registerMemRefPasses();
     mlir::registerConvertToLLVMPass();