From 17053bc11a86f76c97d68ca212c8247b01986774 Mon Sep 17 00:00:00 2001
From: "William S. Moses" <gh@wsmoses.com>
Date: Sun, 5 Oct 2025 12:07:59 -0500
Subject: [PATCH] SROA pass error

---
 src/enzyme_ad/jax/Passes/SROAWrappers.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/enzyme_ad/jax/Passes/SROAWrappers.cpp b/src/enzyme_ad/jax/Passes/SROAWrappers.cpp
index d0986490a..372256c53 100644
--- a/src/enzyme_ad/jax/Passes/SROAWrappers.cpp
+++ b/src/enzyme_ad/jax/Passes/SROAWrappers.cpp
@@ -94,11 +94,16 @@ struct SROAWrappersPass
     pm.addPass(mlir::createConvertNVVMToLLVMPass());
     auto subres = pm.run(mToTranslate);
     if (!subres.succeeded()) {
+      signalPassFailure();
       return;
     }
 
     llvm::LLVMContext llvmCtx;
     auto llvmModule = mlir::translateModuleToLLVMIR(mToTranslate, llvmCtx);
+    if (!llvmModule) {
+      signalPassFailure();
+      return;
+    }
 
     if (dump_prellvm)
       llvm::errs() << "sroa pre llvm\n" << *llvmModule << "\n";