[FnSpecialization] (6/6) Update tests for prior set of changes

Author: bababuck

llvm/test/Transforms/FunctionSpecialization/specialize-chain.ll

@@ -114,19 +114,51 @@ entry:
 ; CHECK-NEXT:    ret i32 0
 ;
 ;
-; CHECK-LABEL: define range(i32 -2147483642, -2147483648) i32 @main() {
+; CHECK-LABEL: define i32 @main() {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[ADD:%.*]] = call i32 @incr.specialized.1(i32 10)
 ; CHECK-NEXT:    [[INT:%.*]] = call i32 @intrinsic(i32 3)
 ; CHECK-NEXT:    [[FWD_UNFOLD:%.*]] = call i32 @forward_unfold(i32 3)
-; CHECK-NEXT:    [[FWD_INNER:%.*]] = call i32 @forward_inner(i32 3)
-; CHECK-NEXT:    [[FWD_OUTER:%.*]] = call i32 @forward_outer(i32 3)
-; CHECK-NEXT:    [[FWD_OUTER1:%.*]] = call i32 @forward_outer(i32 3)
-; CHECK-NEXT:    [[MULTI_CALL:%.*]] = call i32 @multi_call(i32 5)
-; CHECK-NEXT:    ret i32 [[MULTI_CALL]]
+; CHECK-NEXT:    [[FWD_INNER:%.*]] = call i32 @forward_inner.specialized.4(i32 3)
+; CHECK-NEXT:    [[FWD_OUTER:%.*]] = call i32 @forward_outer.specialized.6(i32 3)
+; CHECK-NEXT:    [[FWD_OUTER1:%.*]] = call i32 @forward_outer.specialized.6(i32 3)
+; CHECK-NEXT:    [[MULTI_CALL:%.*]] = call i32 @multi_call.specialized.2(i32 5)
+; CHECK-NEXT:    ret i32 11
 ;
 ;
 ; CHECK-LABEL: define internal i32 @incr.specialized.1(
 ; CHECK-SAME: i32 [[A:%.*]]) {
 ; CHECK-NEXT:    ret i32 poison
 ;
+;
+; CHECK-LABEL: define internal i32 @multi_call.specialized.2(
+; CHECK-SAME: i32 [[A:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    [[CALL:%.*]] = call i32 @incr.specialized.3(i32 5)
+; CHECK-NEXT:    [[MUL_CALL:%.*]] = call i32 @incr.specialized.1(i32 10)
+; CHECK-NEXT:    ret i32 poison
+;
+;
+; CHECK-LABEL: define internal i32 @incr.specialized.3(
+; CHECK-SAME: i32 [[A:%.*]]) {
+; CHECK-NEXT:    ret i32 poison
+;
+;
+; CHECK-LABEL: define internal i32 @forward_inner.specialized.4(
+; CHECK-SAME: i32 [[A:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    [[CALL:%.*]] = call i32 @incr.specialized.5(i32 3)
+; CHECK-NEXT:    ret i32 poison
+;
+;
+; CHECK-LABEL: define internal i32 @incr.specialized.5(
+; CHECK-SAME: i32 [[A:%.*]]) {
+; CHECK-NEXT:    ret i32 poison
+;
+;
+; CHECK-LABEL: define internal i32 @forward_outer.specialized.6(
+; CHECK-SAME: i32 [[A:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    [[CALL:%.*]] = call i32 @forward_inner.specialized.4(i32 3)
+; CHECK-NEXT:    ret i32 poison
+;

llvm/test/Transforms/FunctionSpecialization/track-return.ll

@@ -4,7 +4,7 @@ define i64 @main() {
 ; CHECK:       define i64 @main
 ; CHECK-NEXT:  entry:
 ; CHECK-NEXT:    [[C1:%.*]] = call i64 @foo.specialized.1(i1 true, i64 3, i64 1)
-; CHECK-NEXT:    [[C2:%.*]] = call i64 @foo.specialized.2(i1 false, i64 4, i64 -1)
+; CHECK-NEXT:    [[C2:%.*]] = call i64 @foo.specialized.3(i1 false, i64 4, i64 -1)
 ; CHECK-NEXT:    ret i64 8
 ;
 entry:
@@ -15,27 +15,6 @@ entry:
 }
 
 define internal i64 @foo(i1 %flag, i64 %m, i64 %n) {
-;
-; CHECK:       define internal i64 @foo.specialized.1
-; CHECK-NEXT:  entry:
-; CHECK-NEXT:    br label %plus
-; CHECK:       plus:
-; CHECK-NEXT:    [[N0:%.*]] = call i64 @binop.specialized.4(i64 3, i64 1)
-; CHECK-NEXT:    [[RES0:%.*]] = call i64 @bar.specialized.6(i64 4)
-; CHECK-NEXT:    br label %merge
-; CHECK:       merge:
-; CHECK-NEXT:    ret i64 poison
-;
-; CHECK:       define internal i64 @foo.specialized.2
-; CHECK-NEXT:  entry:
-; CHECK-NEXT:    br label %minus
-; CHECK:       minus:
-; CHECK-NEXT:    [[N1:%.*]] = call i64 @binop.specialized.3(i64 4, i64 -1)
-; CHECK-NEXT:    [[RES1:%.*]] = call i64 @bar.specialized.5(i64 3)
-; CHECK-NEXT:    br label %merge
-; CHECK:       merge:
-; CHECK-NEXT:    ret i64 poison
-;
 entry:
   br i1 %flag, label %plus, label %minus
 
@@ -55,21 +34,61 @@ merge:
 }
 
 define internal i64 @binop(i64 %x, i64 %y) {
+entry:
+  %z = add i64 %x, %y
+  ret i64 %z
+}
+
+define internal i64 @bar(i64 %n) {
+entry:
+  %cmp = icmp sgt i64 %n, 3
+  br i1 %cmp, label %if.then, label %if.else
+
+if.then:
+  %res0 = sdiv i64 %n, 2
+  br label %if.end
+
+if.else:
+  %res1 = mul i64 %n, 2
+  br label %if.end
+
+if.end:
+  %res = phi i64 [ %res0, %if.then ], [ %res1, %if.else]
+  ret i64 %res
+}
+
+;
+; CHECK:       define internal i64 @foo.specialized.1
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    br label %plus
+; CHECK:       plus:
+; CHECK-NEXT:    [[N0:%.*]] = call i64 @binop.specialized.2(i64 3, i64 1)
+; CHECK-NEXT:    [[RES0:%.*]] = call i64 @bar.specialized.6(i64 4)
+; CHECK-NEXT:    br label %merge
+; CHECK:       merge:
+; CHECK-NEXT:    ret i64 poison
+;
 ;
-; CHECK:       define internal i64 @binop.specialized.3
+; CHECK:       define internal i64 @binop.specialized.2
 ; CHECK-NEXT:  entry:
 ; CHECK-NEXT:    ret i64 poison
 ;
+;
+; CHECK:       define internal i64 @foo.specialized.3
+; CHECK-NEXT:  entry:
+; CHECK-NEXT:    br label %minus
+; CHECK:       minus:
+; CHECK-NEXT:    [[N1:%.*]] = call i64 @binop.specialized.4(i64 4, i64 -1)
+; CHECK-NEXT:    [[RES1:%.*]] = call i64 @bar.specialized.5(i64 3)
+; CHECK-NEXT:    br label %merge
+; CHECK:       merge:
+; CHECK-NEXT:    ret i64 poison
+;
+;
 ; CHECK:       define internal i64 @binop.specialized.4
 ; CHECK-NEXT:  entry:
 ; CHECK-NEXT:    ret i64 poison
 ;
-entry:
-  %z = add i64 %x, %y
-  ret i64 %z
-}
-
-define internal i64 @bar(i64 %n) {
 ;
 ; CHECK:       define internal i64 @bar.specialized.5
 ; CHECK-NEXT:  entry:
@@ -87,20 +106,3 @@ define internal i64 @bar(i64 %n) {
 ; CHECK:       if.end:
 ; CHECK-NEXT:    ret i64 poison
 ;
-entry:
-  %cmp = icmp sgt i64 %n, 3
-  br i1 %cmp, label %if.then, label %if.else
-
-if.then:
-  %res0 = sdiv i64 %n, 2
-  br label %if.end
-
-if.else:
-  %res1 = mul i64 %n, 2
-  br label %if.end
-
-if.end:
-  %res = phi i64 [ %res0, %if.then ], [ %res1, %if.else]
-  ret i64 %res
-}
-