1 files changed, 70 insertions, 0 deletions
diff --git a/llvm/test/CodeGen/X86/machine-outliner-noredzone.ll b/llvm/test/CodeGen/X86/machine-outliner-noredzone.ll
new file mode 100644
index 00000000000..94271e30bdf
--- /dev/null
+++ b/llvm/test/CodeGen/X86/machine-outliner-noredzone.ll
@@ -0,0 +1,70 @@
+; RUN: llc -enable-machine-outliner -mtriple=x86_64-apple-darwin < %s | FileCheck %s
+; Ensure that the outliner doesn't outline from any functions that use a redzone.
+
+declare i8* @llvm.stacksave() #1
+declare void @llvm.stackrestore(i8*) #1
+
+; This function has a red zone. We shouldn't outline from it.
+; CHECK-LABEL: doggo
+; CHECK-NOT: OUTLINED
+define void @doggo(i32) #0 {
+  %2 = alloca i32, align 4
+  store i32 %0, i32* %2, align 4
+  %3 = load i32, i32* %2, align 4
+  %4 = add nsw i32 %3, 1
+  store i32 %4, i32* %2, align 4
+  ret void
+}
+
+; Ditto.
+; CHECK-LABEL: pupper
+; CHECK-NOT: OUTLINED
+define void @pupper(i32) #0 {
+  %2 = alloca i32, align 4
+  store i32 %0, i32* %2, align 4
+  %3 = load i32, i32* %2, align 4
+  %4 = add nsw i32 %3, 1
+  store i32 %4, i32* %2, align 4
+  ret void
+}
+
+; This doesn't have a redzone. Outlining is okay.
+; CHECK-LABEL: boofer
+; CHECK: OUTLINED
+define void @boofer(i32) #0 {
+  %2 = alloca i32, align 4
+  %3 = alloca i8*, align 8
+  %4 = alloca i64, align 8
+  store i32 %0, i32* %2, align 4
+  %5 = load i32, i32* %2, align 4
+  %6 = zext i32 %5 to i64
+  %7 = call i8* @llvm.stacksave()
+  store i8* %7, i8** %3, align 8
+  %8 = alloca i32, i64 %6, align 16
+  store i64 %6, i64* %4, align 8
+  %9 = load i8*, i8** %3, align 8
+  call void @llvm.stackrestore(i8* %9)
+  ret void
+}
+
+; Ditto.
+; CHECK-LABEL: shibe
+; CHECK: OUTLINED
+define void @shibe(i32) #0 {
+  %2 = alloca i32, align 4
+  %3 = alloca i8*, align 8
+  %4 = alloca i64, align 8
+  store i32 %0, i32* %2, align 4
+  %5 = load i32, i32* %2, align 4
+  %6 = zext i32 %5 to i64
+  %7 = call i8* @llvm.stacksave()
+  store i8* %7, i8** %3, align 8
+  %8 = alloca i32, i64 %6, align 16
+  store i64 %6, i64* %4, align 8
+  %9 = load i8*, i8** %3, align 8
+  call void @llvm.stackrestore(i8* %9)
+  ret void
+}
+
+attributes #0 = { noinline nounwind optnone ssp uwtable "no-frame-pointer-elim"="true" }
+attributes #1 = { nounwind }
+\ No newline at end of file