From f3b7c5cb9ef443f59593f3dddc7bfd56985bfd0f Mon Sep 17 00:00:00 2001 From: jax authors Date: Tue, 18 Mar 2025 19:22:27 -0700 Subject: [PATCH] Integrate LLVM at llvm/llvm-project@0230d63b4a8b Updates LLVM usage to match [0230d63b4a8b](https://github.com/llvm/llvm-project/commit/0230d63b4a8b) PiperOrigin-RevId: 738222096 --- jaxlib/mosaic/gpu/custom_call.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jaxlib/mosaic/gpu/custom_call.cc b/jaxlib/mosaic/gpu/custom_call.cc index 361c839b6..402e099c8 100644 --- a/jaxlib/mosaic/gpu/custom_call.cc +++ b/jaxlib/mosaic/gpu/custom_call.cc @@ -143,7 +143,7 @@ mlir::FailureOr GetPassPipeline( mlir::memref::registerMemRefPasses(); mlir::registerConvertToLLVMPass(); mlir::registerGPUPasses(); - mlir::registerGpuLaunchSinkIndexComputations(); + mlir::registerGpuLaunchSinkIndexComputationsPass(); mosaic::gpu::registerGpuLaunchLoweringPass(); mosaic::gpu::registerConvertGpuToLLVMPass(); mosaic::gpu::registerByvalInsertionPass();