Fix mem2reg on input fn arg (#1442)

EnzymeAD · Sep 19, 2023 · e8e4188 · e8e4188
1 parent 3870f89
commit e8e4188
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -2318,6 +2318,9 @@ Function *GetFunctionFromValue(Value *fn) {
             continue;
           }
 
+          if (isa<LoadInst>(cur))
+            continue;
+
           if (!cur->mayWriteToMemory() && cur->getType()->isVoidTy())
             continue;
 

diff --git a/enzyme/test/Enzyme/ReverseMode/mem2regfn.ll b/enzyme/test/Enzyme/ReverseMode/mem2regfn.ll
@@ -0,0 +1,18 @@
+; RUN: if [ %llvmver -lt 16 ]; then %opt < %s %loadEnzyme -enzyme-preopt=false -enzyme -S | FileCheck %s; fi
+; RUN: %opt < %s %newLoadEnzyme -enzyme-preopt=false -passes="enzyme" -S | FileCheck %s
+
+define double @_Z3food(double %0) {
+  ret double %0
+}
+
+define i32 @main()  {
+  %a5 = alloca i8*, align 8
+  store i8* bitcast (double (double)* @_Z3food to i8*), i8** %a5, align 8
+  %a17 = load i8*, i8** %a5, align 8
+  %q = call double (...) @__enzyme_autodiff(i8* %a17, metadata !"enzyme_dup", double 1.0, double 1.0)
+  ret i32 0
+}
+
+declare double @__enzyme_autodiff(...)
+
+; CHECK: define internal void @diffe_Z3food