[llvm] 628c537 - [MTE] Add test that stack tagging does not mess up stack coloring.

[Florian Mayer via llvm-commits]

Author: Florian Mayer
Date: 2022-03-14T13:36:21-07:00
New Revision: 628c537b3203bf2ff08e9ca64dbf7c9c85bf4f9f

URL: https://github.com/llvm/llvm-project/commit/628c537b3203bf2ff08e9ca64dbf7c9c85bf4f9f
DIFF: https://github.com/llvm/llvm-project/commit/628c537b3203bf2ff08e9ca64dbf7c9c85bf4f9f.diff

LOG: [MTE] Add test that stack tagging does not mess up stack coloring.

Reviewed By: eugenis

Differential Revision: https://reviews.llvm.org/D121433

Added: 
    llvm/test/CodeGen/AArch64/stack-tagging-stack-coloring.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/CodeGen/AArch64/stack-tagging-stack-coloring.ll b/llvm/test/CodeGen/AArch64/stack-tagging-stack-coloring.ll
new file mode 100644
index 0000000000000..aa38bb39bd347
--- /dev/null
+++ b/llvm/test/CodeGen/AArch64/stack-tagging-stack-coloring.ll
@@ -0,0 +1,41 @@
+; Test that storage for allocas with disjoint lifetimes is reused with stack
+; tagging.
+
+; RUN: opt -S -aarch64-stack-tagging %s -o - | \
+; RUN:   llc -no-stack-coloring=false -o - | \
+; RUN:   FileCheck %s --check-prefix=COLOR
+; RUN: opt -S -aarch64-stack-tagging %s -o - | \
+; RUN:   llc -no-stack-coloring=true -o - | \
+; RUN:   FileCheck %s --check-prefix=NOCOLOR
+
+target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
+target triple = "aarch64-unknown-linux-android29"
+
+; COLOR: sub	sp, sp, #192
+; NOCOLOR: sub	sp, sp, #320
+
+define i32 @myCall_w2(i32 %in) sanitize_hwaddress {
+entry:
+  %a = alloca [17 x i8*], align 8
+  %a2 = alloca [16 x i8*], align 8
+  %b = bitcast [17 x i8*]* %a to i8*
+  %b2 = bitcast [16 x i8*]* %a2 to i8*
+  call void @llvm.lifetime.start.p0i8(i64 136, i8* %b)
+  %t1 = call i32 @foo(i32 %in, i8* %b)
+  %t2 = call i32 @foo(i32 %in, i8* %b)
+  call void @llvm.lifetime.end.p0i8(i64 136, i8* %b)
+  call void @llvm.lifetime.start.p0i8(i64 128, i8* %b2)
+  %t3 = call i32 @foo(i32 %in, i8* %b2)
+  %t4 = call i32 @foo(i32 %in, i8* %b2)
+  call void @llvm.lifetime.end.p0i8(i64 128, i8* %b2)
+  %t5 = add i32 %t1, %t2
+  %t6 = add i32 %t3, %t4
+  %t7 = add i32 %t5, %t6
+  ret i32 %t7
+}
+
+declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture) nounwind
+
+declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture) nounwind
+
+declare i32 @foo(i32, i8*)