Fix named bcloader (#2150)

EnzymeAD · Nov 4, 2024 · e5a36ad · e5a36ad
1 parent b38b240
commit e5a36ad
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/enzyme/BCLoad/BCLoader.cpp b/enzyme/BCLoad/BCLoader.cpp
@@ -75,8 +75,7 @@ bool provideDefinitions(Module &M, std::set<std::string> ignoreFunctions,
 
   for (auto &&[realname, F] : name_rewrites) {
     auto decl = M.getOrInsertFunction(realname, F->getFunctionType());
-    auto entry = BasicBlock::Create(F->getContext(), "entry",
-                                    cast<Function>(decl.getCallee()));
+    auto entry = BasicBlock::Create(F->getContext(), "entry", F);
     IRBuilder<> B(entry);
     SmallVector<Value *, 1> vals;
     for (auto &arg : F->args())