xref: /llvm-project/llvm/test/Transforms/IndVarSimplify/pr55925.ll (revision 4b22a923c4bfd0aa1d483149f84b6787263c2d76)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -S -passes=indvars < %s | FileCheck %s
3
4; This tests the case where a terminator can be modeled by SCEV,
5; because it has a returned attribute.
6
7target datalayout = "e-m:o-i64:64-i128:128-n32:64-S128"
8
9declare i32 @foo(i32)
10
11define void @test(ptr %p) personality ptr undef {
12; CHECK-LABEL: @test(
13; CHECK-NEXT:  entry:
14; CHECK-NEXT:    br label [[LOOP:%.*]]
15; CHECK:       loop:
16; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ [[INDVARS_IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ], [ 0, [[ENTRY:%.*]] ]
17; CHECK-NEXT:    [[TMP0:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
18; CHECK-NEXT:    [[RES:%.*]] = invoke i32 @foo(i32 returned [[TMP0]])
19; CHECK-NEXT:            to label [[LOOP_LATCH]] unwind label [[EXIT:%.*]]
20; CHECK:       loop.latch:
21; CHECK-NEXT:    [[TMP1:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
22; CHECK-NEXT:    [[TMP2:%.*]] = call i32 @foo(i32 [[TMP1]])
23; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add i64 [[INDVARS_IV]], 1
24; CHECK-NEXT:    br label [[LOOP]]
25; CHECK:       exit:
26; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
27; CHECK-NEXT:            cleanup
28; CHECK-NEXT:    ret void
29;
30entry:
31  br label %loop
32
33loop:
34  %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
35  %res = invoke i32 @foo(i32 returned %iv)
36  to label %loop.latch unwind label %exit
37
38loop.latch:
39  %ext = zext i32 %iv to i64
40  %tmp5 = getelementptr inbounds i8, ptr %p, i64 %ext
41  %iv.next = add nuw i32 %iv, 1
42  call i32 @foo(i32 %res)
43  br label %loop
44
45exit:
46  %lp = landingpad { ptr, i32 }
47  cleanup
48  ret void
49}
50
51define void @test_critedge(i1 %c, ptr %p) personality ptr undef {
52; CHECK-LABEL: @test_critedge(
53; CHECK-NEXT:  entry:
54; CHECK-NEXT:    br label [[LOOP:%.*]]
55; CHECK:       loop:
56; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ [[INDVARS_IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ], [ 0, [[ENTRY:%.*]] ]
57; CHECK-NEXT:    br i1 [[C:%.*]], label [[LOOP_INVOKE:%.*]], label [[LOOP_OTHER:%.*]]
58; CHECK:       loop.invoke:
59; CHECK-NEXT:    [[TMP0:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
60; CHECK-NEXT:    [[TMP1:%.*]] = trunc nuw i64 [[INDVARS_IV]] to i32
61; CHECK-NEXT:    [[RES:%.*]] = invoke i32 @foo(i32 returned [[TMP0]])
62; CHECK-NEXT:            to label [[LOOP_LATCH]] unwind label [[EXIT:%.*]]
63; CHECK:       loop.other:
64; CHECK-NEXT:    br label [[LOOP_LATCH]]
65; CHECK:       loop.latch:
66; CHECK-NEXT:    [[PHI:%.*]] = phi i32 [ [[TMP1]], [[LOOP_INVOKE]] ], [ 0, [[LOOP_OTHER]] ]
67; CHECK-NEXT:    [[TMP2:%.*]] = call i32 @foo(i32 [[PHI]])
68; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add i64 [[INDVARS_IV]], 1
69; CHECK-NEXT:    br label [[LOOP]]
70; CHECK:       exit:
71; CHECK-NEXT:    [[LP:%.*]] = landingpad { ptr, i32 }
72; CHECK-NEXT:            cleanup
73; CHECK-NEXT:    ret void
74;
75entry:
76  br label %loop
77
78loop:
79  %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop.latch ]
80  br i1 %c, label %loop.invoke, label %loop.other
81
82loop.invoke:
83  %res = invoke i32 @foo(i32 returned %iv)
84  to label %loop.latch unwind label %exit
85
86loop.other:
87  br label %loop.latch
88
89loop.latch:
90  %phi = phi i32 [ %res, %loop.invoke ], [ 0, %loop.other ]
91  %ext = zext i32 %iv to i64
92  %tmp5 = getelementptr inbounds i8, ptr %p, i64 %ext
93  %iv.next = add nuw i32 %iv, 1
94  call i32 @foo(i32 %phi)
95  br label %loop
96
97exit:
98  %lp = landingpad { ptr, i32 }
99  cleanup
100  ret void
101}
102