From 44f6c711dd7b7cc3c0caaf1913fd3e7dc9646133 Mon Sep 17 00:00:00 2001
From: Dan <tthsqe12@gmail.com>
Date: Mon, 9 Oct 2023 20:43:12 -0400
Subject: [PATCH] ensure that double stores to tape do not incure additional
 mallocs

---
 enzyme/test/Enzyme/ReverseMode/nomallocs.ll | 73 +++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 enzyme/test/Enzyme/ReverseMode/nomallocs.ll

diff --git a/enzyme/test/Enzyme/ReverseMode/nomallocs.ll b/enzyme/test/Enzyme/ReverseMode/nomallocs.ll
new file mode 100644
index 000000000000..68214696be86
--- /dev/null
+++ b/enzyme/test/Enzyme/ReverseMode/nomallocs.ll
@@ -0,0 +1,73 @@
+; RUN: if [ %llvmver -lt 16 ]; then %opt < %s %loadEnzyme -enzyme-preopt=false -enzyme -mem2reg -instsimplify -simplifycfg -S | FileCheck %s; fi
+; RUN: %opt < %s %newLoadEnzyme -enzyme-preopt=false -passes="enzyme,function(mem2reg,instsimplify,%simplifycfg)" -S | FileCheck %s
+
+; the f function is simple enough that no calls to malloc should be required here
+
+; ModuleID = 'nomallocs.c'
+source_filename = "nomallocs.c"
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+@enzyme_allocated = external local_unnamed_addr global i32, align 4
+@enzyme_tape = external local_unnamed_addr global i32, align 4
+
+; Function Attrs: mustprogress nofree nounwind willreturn uwtable
+define dso_local void @f(ptr nocapture noundef writeonly %y, ptr nocapture noundef readonly %x) #0 {
+entry:
+  %0 = load double, ptr %x, align 8, !tbaa !5
+  %call = tail call double @sin(double noundef %0) #5
+  store double %call, ptr %y, align 8, !tbaa !5
+  ret void
+}
+
+; Function Attrs: mustprogress nofree nounwind willreturn writeonly
+declare double @sin(double noundef) local_unnamed_addr #1
+
+; Function Attrs: nounwind uwtable
+define dso_local void @f_b(ptr noundef %y, ptr noundef %y_b, ptr noundef %x, ptr noundef %x_b) local_unnamed_addr #2 {
+entry:
+  %tape = alloca [100 x i8], align 16
+  call void @llvm.lifetime.start.p0(i64 100, ptr nonnull %tape) #5
+  %0 = load i32, ptr @enzyme_allocated, align 4, !tbaa !9
+  %1 = load i32, ptr @enzyme_tape, align 4, !tbaa !9
+  %call = call ptr (ptr, ...) @__enzyme_augmentfwd(ptr noundef nonnull @f, ptr noundef %y, ptr noundef %y_b, ptr noundef %x, ptr noundef %x_b, i32 noundef %0, i64 noundef 100, i32 noundef %1, ptr noundef nonnull %tape) #5
+  %2 = load i32, ptr @enzyme_allocated, align 4, !tbaa !9
+  %3 = load i32, ptr @enzyme_tape, align 4, !tbaa !9
+  call void (ptr, ...) @__enzyme_reverse(ptr noundef nonnull @f, ptr noundef %x, ptr noundef %y_b, ptr noundef %x, ptr noundef %x_b, i32 noundef %2, i64 noundef 100, i32 noundef %3, ptr noundef nonnull %tape) #5
+  call void @llvm.lifetime.end.p0(i64 100, ptr nonnull %tape) #5
+  ret void
+}
+
+; Function Attrs: argmemonly mustprogress nocallback nofree nosync nounwind willreturn
+declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture) #3
+
+declare ptr @__enzyme_augmentfwd(ptr noundef, ...) local_unnamed_addr #4
+
+declare void @__enzyme_reverse(ptr noundef, ...) local_unnamed_addr #4
+
+; Function Attrs: argmemonly mustprogress nocallback nofree nosync nounwind willreturn
+declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture) #3
+
+attributes #0 = { mustprogress nofree nounwind willreturn uwtable "frame-pointer"="none" "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
+attributes #1 = { mustprogress nofree nounwind willreturn writeonly "frame-pointer"="none" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
+attributes #2 = { nounwind uwtable "frame-pointer"="none" "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
+attributes #3 = { argmemonly mustprogress nocallback nofree nosync nounwind willreturn }
+attributes #4 = { "frame-pointer"="none" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
+attributes #5 = { nounwind }
+
+!llvm.module.flags = !{!0, !1, !2, !3}
+!llvm.ident = !{!4}
+
+!0 = !{i32 1, !"wchar_size", i32 4}
+!1 = !{i32 7, !"PIC Level", i32 2}
+!2 = !{i32 7, !"PIE Level", i32 2}
+!3 = !{i32 7, !"uwtable", i32 2}
+!4 = !{!"clang version 15.0.7"}
+!5 = !{!6, !6, i64 0}
+!6 = !{!"double", !7, i64 0}
+!7 = !{!"omnipotent char", !8, i64 0}
+!8 = !{!"Simple C/C++ TBAA"}
+!9 = !{!10, !10, i64 0}
+!10 = !{!"int", !7, i64 0}
+
+; CHECK-NOT: malloc