Add debug info for binary lowering

EnzymeAD · Jan 9, 2025 · 058fa1e · 058fa1e
1 parent aa23e9f
commit 058fa1e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/LowerKernel.cpp b/src/enzyme_ad/jax/Passes/LowerKernel.cpp
@@ -561,7 +561,7 @@ CallInfo CompileKernel(SymbolTableCollection &symbolTable, mlir::Location loc,
             builder.getStringAttr(value + '\0'));
       }
 
-      LLVM::GlobalOp binary;
+      LLVM::GlobalOp binary = nullptr;
       submod.walk([&](gpu::BinaryOp op) {
         gpu::ObjectAttr object = getSelectedObject(op);
         auto value = object.getObject().getValue();
@@ -583,6 +583,11 @@ CallInfo CompileKernel(SymbolTableCollection &symbolTable, mlir::Location loc,
             builder.getContext(), mlir::LLVM::UnnamedAddr::None));
         op.erase();
       });
+      if (!binary) {
+        llvm::errs() << "could not find binary object in submod:\n"
+                     << *submod << "\n";
+        assert(binary);
+      }
 
       {
         auto blk = new Block();