Handle batch ffi abi (#1390)

EnzymeAD · Aug 29, 2023 · 1d62f3f · 1d62f3f
1 parent 8a24100
commit 1d62f3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/enzyme/Enzyme/AdjointGenerator.h b/enzyme/Enzyme/AdjointGenerator.h
@@ -7507,7 +7507,7 @@ class AdjointGenerator
                         : BuilderZ.CreateExtractValue(augmentcall,
                                                       {(unsigned)drval},
                                                       call.getName() + "'ac");
-            assert(newip->getType() == call.getType());
+            assert(newip->getType() == placeholder->getType());
             placeholder->replaceAllUsesWith(newip);
             if (placeholder == &*BuilderZ.GetInsertPoint()) {
               BuilderZ.SetInsertPoint(placeholder->getNextNode());