1; RUN: opt < %s -S -loop-unroll -unroll-runtime -unroll-count=2 -unroll-runtime-epilog=true | FileCheck %s -check-prefix=EPILOG
2; RUN: opt < %s -S -loop-unroll -unroll-runtime -unroll-count=2 -unroll-runtime-epilog=false | FileCheck %s -check-prefix=PROLOG
3
4; RUN: opt < %s -S -passes='require<opt-remark-emit>,unroll' -unroll-runtime -unroll-count=2 -unroll-runtime-epilog=true | FileCheck %s -check-prefix=EPILOG
5; RUN: opt < %s -S -passes='require<opt-remark-emit>,unroll' -unroll-runtime -unroll-count=2 -unroll-runtime-epilog=false | FileCheck %s -check-prefix=PROLOG
6
7; This tests that setting the unroll count works
8
9
10; EPILOG: for.body.preheader:
11; EPILOG:   br i1 %1, label %for.end.loopexit.unr-lcssa, label %for.body.preheader.new, !dbg [[PH_LOC:![0-9]+]]
12; EPILOG: for.body:
13; EPILOG:   br i1 %niter.ncmp.1, label %for.end.loopexit.unr-lcssa.loopexit, label %for.body, !dbg [[BODY_LOC:![0-9]+]]
14; EPILOG-NOT: br i1 %niter.ncmp.2, label %for.end.loopexit{{.*}}, label %for.body
15; EPILOG: for.body.epil.preheader:
16; EPILOG:   br label %for.body.epil, !dbg [[BODY_LOC]]
17; EPILOG: for.body.epil:
18; EPILOG:   br label %for.end.loopexit.epilog-lcssa, !dbg [[BODY_LOC]]
19; EPILOG: for.end.loopexit:
20; EPILOG:   br label %for.end, !dbg [[EXIT_LOC:![0-9]+]]
21
22; EPILOG-DAG: [[PH_LOC]] = !DILocation(line: 101, column: 1, scope: !{{.*}})
23; EPILOG-DAG: [[BODY_LOC]] = !DILocation(line: 102, column: 1, scope: !{{.*}})
24; EPILOG-DAG: [[EXIT_LOC]] = !DILocation(line: 103, column: 1, scope: !{{.*}})
25
26; PROLOG: for.body.preheader:
27; PROLOG:   br {{.*}} label %for.body.prol.preheader, label %for.body.prol.loopexit, !dbg [[PH_LOC:![0-9]+]]
28; PROLOG: for.body.prol:
29; PROLOG:   br label %for.body.prol.loopexit, !dbg [[BODY_LOC:![0-9]+]]
30; PROLOG: for.body.prol.loopexit:
31; PROLOG:   br {{.*}} label %for.end.loopexit, label %for.body.preheader.new, !dbg [[PH_LOC]]
32; PROLOG: for.body:
33; PROLOG:   br i1 %exitcond.1, label %for.end.loopexit.unr-lcssa, label %for.body, !dbg [[BODY_LOC]]
34; PROLOG-NOT: br i1 %exitcond.4, label %for.end.loopexit{{.*}}, label %for.body
35
36; PROLOG-DAG: [[PH_LOC]] = !DILocation(line: 101, column: 1, scope: !{{.*}})
37; PROLOG-DAG: [[BODY_LOC]] = !DILocation(line: 102, column: 1, scope: !{{.*}})
38
39define i32 @test(i32* nocapture %a, i32 %n) nounwind uwtable readonly !dbg !6 {
40entry:
41  %cmp1 = icmp eq i32 %n, 0, !dbg !7
42  br i1 %cmp1, label %for.end, label %for.body, !dbg !7
43
44for.body:                                         ; preds = %for.body, %entry
45  %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
46  %sum.02 = phi i32 [ %add, %for.body ], [ 0, %entry ]
47  %arrayidx = getelementptr inbounds i32, i32* %a, i64 %indvars.iv, !dbg !8
48  %0 = load i32, i32* %arrayidx, align 4, !dbg !8
49  %add = add nsw i32 %0, %sum.02, !dbg !8
50  %indvars.iv.next = add i64 %indvars.iv, 1, !dbg !9
51  %lftr.wideiv = trunc i64 %indvars.iv.next to i32, !dbg !9
52  %exitcond = icmp eq i32 %lftr.wideiv, %n, !dbg !9
53  br i1 %exitcond, label %for.end, label %for.body, !dbg !9
54
55for.end:                                          ; preds = %for.body, %entry
56  %sum.0.lcssa = phi i32 [ 0, %entry ], [ %add, %for.body ]
57  ret i32 %sum.0.lcssa, !dbg !10
58}
59
60!llvm.module.flags = !{!0, !1, !2}
61!llvm.dbg.cu = !{!11}
62!0 = !{i32 2, !"Dwarf Version", i32 4}
63!1 = !{i32 2, !"Debug Info Version", i32 3}
64!2 = !{i32 1, !"PIC Level", i32 2}
65
66!3 = !{}
67!4 = !DISubroutineType(types: !3)
68!5 = !DIFile(filename: "test.cpp", directory: "/tmp")
69!6 = distinct !DISubprogram(name: "test", scope: !5, file: !5, line: 99, type: !4, isLocal: false, isDefinition: true, scopeLine: 100, flags: DIFlagPrototyped, isOptimized: false, unit: !11, retainedNodes: !3)
70!7 = !DILocation(line: 100, column: 1, scope: !6)
71!8 = !DILocation(line: 101, column: 1, scope: !6)
72!9 = !DILocation(line: 102, column: 1, scope: !6)
73!10 = !DILocation(line: 103, column: 1, scope: !6)
74!11 = distinct !DICompileUnit(language: DW_LANG_C99, producer: "clang",
75                             file: !5,
76                             isOptimized: true, flags: "-O2",
77                             splitDebugFilename: "abc.debug", emissionKind: 2)
78