diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp
index f64fd63829..108f292716 100644
--- a/deps/ReactantExtra/API.cpp
+++ b/deps/ReactantExtra/API.cpp
@@ -1076,11 +1076,10 @@ extern "C" void InitializePasses(MlirDialectRegistry creg) {
   enzyme::registerenzymexlaPasses();
 
   // Register the standard passes we want.
-  mlir::registerCSEPass();
+  mlir::registerTransformsPasses();
   mlir::registerLowerAffinePass();
   mlir::registerSCCPPass();
   mlir::registerInlinerPass();
-  mlir::registerCanonicalizerPass();
   mlir::registerSymbolDCEPass();
   mlir::registerLoopInvariantCodeMotionPass();
   mlir::registerConvertSCFToOpenMPPass();
diff --git a/src/Compiler.jl b/src/Compiler.jl
index d1d37f3455..1866e60a7f 100644
--- a/src/Compiler.jl
+++ b/src/Compiler.jl
@@ -748,7 +748,11 @@ function compile_mlir!(
 
     if backend == "cpu" || backend == "tpu"
         kern = "lower-kernel{backend=cpu},canonicalize"
-        jit = "lower-jit{openmp=true backend=cpu},symbol-dce"
+        if backend == "tpu"
+            jit = "lower-jit{openmp=true backend=cpu},symbol-dce,strip-debuginfo"
+        else
+            jit = "lower-jit{openmp=true backend=cpu},symbol-dce"
+        end
     elseif DEBUG_KERNEL[]
         curesulthandler = dlsym(
             Reactant_jll.libReactantExtra_handle, "ReactantHandleCuResult"