fix

EnzymeAD · Dec 24, 2024 · c4588eb · c4588eb
1 parent 5ebc464
commit c4588eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/enzyme_ad/jax/Passes/LowerKernel.cpp b/src/enzyme_ad/jax/Passes/LowerKernel.cpp
@@ -557,7 +557,7 @@ void *CompileKernel(SymbolTableCollection &symbolTable, mlir::Location loc,
       OpBuilder builder(submod);
 
       SymbolTable st2(submod);
-      auto print2 = st2.lookup<LLVM::LLVMFuncOp>(submod);
+      auto print2 = st2.lookup<LLVM::LLVMFuncOp>("printf");
 
       builder.setInsertionPointToStart(&submod.getBodyRegion().front());
       auto ptrty = LLVM::LLVMPointerType::get(builder.getContext());