[llvm] 2a1ac2e - [FuncSpec] Add force flag to test case to trigger the transform. NFC.

[Sjoerd Meijer via llvm-commits]

Author: Sjoerd Meijer
Date: 2021-09-16T17:48:13+01:00
New Revision: 2a1ac2e318c60955aecefb493e4707f0d609c35d

URL: https://github.com/llvm/llvm-project/commit/2a1ac2e318c60955aecefb493e4707f0d609c35d
DIFF: https://github.com/llvm/llvm-project/commit/2a1ac2e318c60955aecefb493e4707f0d609c35d.diff

LOG: [FuncSpec] Add force flag to test case to trigger the transform. NFC.

Added: 
    

Modified: 
    llvm/test/Transforms/FunctionSpecialization/function-specialization-constant-expression.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/FunctionSpecialization/function-specialization-constant-expression.ll b/llvm/test/Transforms/FunctionSpecialization/function-specialization-constant-expression.ll
index b38391d7e972d..d0a912447822e 100644
--- a/llvm/test/Transforms/FunctionSpecialization/function-specialization-constant-expression.ll
+++ b/llvm/test/Transforms/FunctionSpecialization/function-specialization-constant-expression.ll
@@ -1,16 +1,28 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+
 ; Test function specialization wouldn't crash due to constant expression.
 ; Note that this test case shows that function specialization pass would
 ; transform the function even if no specialization happened.
-; RUN: opt -function-specialization -S < %s | FileCheck %s
 
-; CHECK: plus:
-; CHECK-NEXT:  %{{.*}} = call i64 @func2(i64* getelementptr inbounds (%struct, %struct* @Global, i32 0, i32 3))
-; CHECK: minus:
-; CHECK-NEXT: %{{.*}} = call i64 @func2(i64* getelementptr inbounds (%struct, %struct* @Global, i32 0, i32 4))
+; RUN: opt -function-specialization -force-function-specialization -S < %s | FileCheck %s
 
 %struct = type { i8, i16, i32, i64, i64}
 @Global = internal constant %struct {i8 0, i16 1, i32 2, i64 3, i64 4}
+
 define internal i64 @zoo(i1 %flag) {
+; CHECK-LABEL: @zoo(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    br i1 [[FLAG:%.*]], label [[PLUS:%.*]], label [[MINUS:%.*]]
+; CHECK:       plus:
+; CHECK-NEXT:    [[TMP0:%.*]] = call i64 @func2.1(i64* getelementptr inbounds ([[STRUCT:%.*]], %struct* @Global, i32 0, i32 3))
+; CHECK-NEXT:    br label [[MERGE:%.*]]
+; CHECK:       minus:
+; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @func2.2(i64* getelementptr inbounds ([[STRUCT]], %struct* @Global, i32 0, i32 4))
+; CHECK-NEXT:    br label [[MERGE]]
+; CHECK:       merge:
+; CHECK-NEXT:    [[TMP2:%.*]] = phi i64 [ [[TMP0]], [[PLUS]] ], [ [[TMP1]], [[MINUS]] ]
+; CHECK-NEXT:    ret i64 [[TMP2]]
+;
 entry:
   br i1 %flag, label %plus, label %minus
 
@@ -30,20 +42,46 @@ merge:
 }
 
 define internal i64 @func2(i64 *%x) {
+; CHECK-LABEL: @func2(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[VAL:%.*]] = ptrtoint i64* [[X:%.*]] to i64
+; CHECK-NEXT:    ret i64 [[VAL]]
+;
 entry:
   %val = ptrtoint i64* %x to i64
   ret i64 %val
 }
 
 define internal i64 @func(i64 *%x, i64 (i64*)* %binop) {
+; CHECK-LABEL: @func(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    [[TMP0:%.*]] = call i64 [[BINOP:%.*]](i64* [[X:%.*]])
+; CHECK-NEXT:    ret i64 [[TMP0]]
+;
 entry:
   %tmp0 = call i64 %binop(i64* %x)
   ret i64 %tmp0
 }
 
 define i64 @main() {
-    %1 = call i64 @zoo(i1 0)
-    %2 = call i64 @zoo(i1 1)
-    %3 = add i64 %1, %2
-    ret i64 %3
-}
\ No newline at end of file
+; CHECK-LABEL: @main(
+; CHECK-NEXT:    [[TMP1:%.*]] = call i64 @zoo(i1 false)
+; CHECK-NEXT:    [[TMP2:%.*]] = call i64 @zoo(i1 true)
+; CHECK-NEXT:    [[TMP3:%.*]] = add i64 [[TMP1]], [[TMP2]]
+; CHECK-NEXT:    ret i64 [[TMP3]]
+;
+  %1 = call i64 @zoo(i1 0)
+  %2 = call i64 @zoo(i1 1)
+  %3 = add i64 %1, %2
+  ret i64 %3
+}
+
+; CHECK-LABEL: define internal i64 @func2.1(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    ret i64 ptrtoint (i64* getelementptr inbounds (%struct, %struct* @Global, i32 0, i32 3) to i64)
+; CHECK-NEXT:  }
+
+; CHECK-LABEL: define internal i64 @func2.2(
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    ret i64 ptrtoint (i64* getelementptr inbounds (%struct, %struct* @Global, i32 0, i32 4) to i64)
+; CHECK-NEXT:  }