xref: /llvm-project/llvm/test/Transforms/CodeExtractor/LoopExtractor.ll (revision 0ca8ddd84d416b2b2b831fbc75ac8dbb77a12928) 
1; RUN: opt < %s -passes='function(break-crit-edges,loop-simplify),loop-extract' -S | FileCheck %s
2
3; This function has 2 simple loops and they should be extracted into 2 new functions.
4define void @test3() {
5; CHECK-LABEL: @test3(
6; CHECK-NEXT:  entry:
7; CHECK-NEXT:    br label %codeRepl1
8; CHECK:       codeRepl1:
9; CHECK-NEXT:    call void @test3.loop.0()
10; CHECK-NEXT:    br label %loop.0.loop.1_crit_edge
11; CHECK:       loop.0.loop.1_crit_edge:
12; CHECK-NEXT:    br label %codeRepl
13; CHECK:       codeRepl:
14; CHECK-NEXT:    call void @test3.loop.1()
15; CHECK-NEXT:    br label %exit
16; CHECK:       exit:
17; CHECK-NEXT:    ret void
18
19entry:
20  br label %loop.0
21
22loop.0:                                           ; preds = %loop.0, %entry
23  %index.0 = phi i32 [ 10, %entry ], [ %next.0, %loop.0 ]
24  tail call void @foo()
25  %next.0 = add nsw i32 %index.0, -1
26  %repeat.0 = icmp sgt i32 %index.0, 1
27  br i1 %repeat.0, label %loop.0, label %loop.1
28
29loop.1:                                           ; preds = %loop.0, %loop.1
30  %index.1 = phi i32 [ %next.1, %loop.1 ], [ 10, %loop.0 ]
31  tail call void @foo()
32  %next.1 = add nsw i32 %index.1, -1
33  %repeat.1 = icmp sgt i32 %index.1, 1
34  br i1 %repeat.1, label %loop.1, label %exit
35
36exit:                                             ; preds = %loop.1
37  ret void
38}
39
40declare void @foo()
41
42; CHECK-LABEL: define internal void @test3.loop.1()
43; CHECK-NEXT:  newFuncRoot:
44; CHECK-NEXT:    br label %loop.1
45; CHECK:       loop.1:
46; CHECK-NEXT:    %index.1 = phi i32 [ %next.1, %loop.1.loop.1_crit_edge ], [ 10, %newFuncRoot ]
47; CHECK-NEXT:    tail call void @foo()
48; CHECK-NEXT:    %next.1 = add nsw i32 %index.1, -1
49; CHECK-NEXT:    %repeat.1 = icmp sgt i32 %index.1, 1
50; CHECK-NEXT:    br i1 %repeat.1, label %loop.1.loop.1_crit_edge, label %exit.exitStub
51; CHECK:       loop.1.loop.1_crit_edge:
52; CHECK-NEXT:    br label %loop.1
53; CHECK:       exit.exitStub:
54; CHECK-NEXT:    ret void
55
56; CHECK-LABEL: define internal void @test3.loop.0()
57; CHECK-NEXT:  newFuncRoot:
58; CHECK-NEXT:    br label %loop.0
59; CHECK:       loop.0:
60; CHECK-NEXT:    %index.0 = phi i32 [ 10, %newFuncRoot ], [ %next.0, %loop.0.loop.0_crit_edge ]
61; CHECK-NEXT:    tail call void @foo()
62; CHECK-NEXT:    %next.0 = add nsw i32 %index.0, -1
63; CHECK-NEXT:    %repeat.0 = icmp sgt i32 %index.0, 1
64; CHECK-NEXT:    br i1 %repeat.0, label %loop.0.loop.0_crit_edge, label %loop.0.loop.1_crit_edge.exitStub
65; CHECK:       loop.0.loop.0_crit_edge:
66; CHECK-NEXT:    br label %loop.0
67; CHECK:       loop.0.loop.1_crit_edge.exitStub:
68; CHECK-NEXT:    ret void
69