Pass if is unnecessary (#2207)

EnzymeAD · Dec 26, 2024 · 5954062 · 5954062
1 parent 1ca3ceb
commit 5954062
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/enzyme/Enzyme/CallDerivatives.cpp b/enzyme/Enzyme/CallDerivatives.cpp
@@ -30,7 +30,7 @@ using namespace llvm;
 
 extern "C" {
 void (*EnzymeShadowAllocRewrite)(LLVMValueRef, void *, LLVMValueRef, uint64_t,
-                                 LLVMValueRef) = nullptr;
+                                 LLVMValueRef, uint8_t) = nullptr;
 }
 
 void AdjointGenerator::handleMPI(llvm::CallInst &call, llvm::Function *called,
@@ -3062,9 +3062,12 @@ bool AdjointGenerator::handleKnownCallDerivatives(
                 if (funcName == "julia.gc_alloc_obj" ||
                     funcName == "jl_gc_alloc_typed" ||
                     funcName == "ijl_gc_alloc_typed") {
-                  if (EnzymeShadowAllocRewrite)
+                  if (EnzymeShadowAllocRewrite) {
+                    bool used = unnecessaryInstructions.find(&call) ==
+                                unnecessaryInstructions.end();
                     EnzymeShadowAllocRewrite(wrap(anti), gutils, wrap(&call),
-                                             idx, wrap(prev));
+                                             idx, wrap(prev), used);
+                  }
                 }
               }
               if (Mode == DerivativeMode::ReverseModeCombined ||
@@ -3249,9 +3252,12 @@ bool AdjointGenerator::handleKnownCallDerivatives(
           if (funcName == "julia.gc_alloc_obj" ||
               funcName == "jl_gc_alloc_typed" ||
               funcName == "ijl_gc_alloc_typed") {
-            if (EnzymeShadowAllocRewrite)
+            if (EnzymeShadowAllocRewrite) {
+              bool used = unnecessaryInstructions.find(&call) ==
+                          unnecessaryInstructions.end();
               EnzymeShadowAllocRewrite(wrap(CI), gutils, wrap(&call), idx,
-                                       wrap(prev));
+                                       wrap(prev), used);
+            }
           }
           idx++;
           prev = CI;