| ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5 | |
| ; RUN: opt -passes=loop-unroll -S < %s | FileCheck %s | |
| define void @full_and_enable(ptr %p, i32 %n) { | |
| ; CHECK-LABEL: define void @full_and_enable( | |
| ; CHECK-SAME: ptr [[P:%.*]], i32 [[N:%.*]]) { | |
| ; CHECK-NEXT: [[ENTRY:.*:]] | |
| ; CHECK-NEXT: [[MASK:%.*]] = and i32 [[N]], 1 | |
| ; CHECK-NEXT: [[BOUND:%.*]] = add nuw i32 [[MASK]], 1 | |
| ; CHECK-NEXT: br label %[[LOOP:.*]] | |
| ; CHECK: [[LOOP]]: | |
| ; CHECK-NEXT: store i32 0, ptr [[P]], align 4 | |
| ; CHECK-NEXT: [[CMP:%.*]] = icmp ult i32 1, [[BOUND]] | |
| ; CHECK-NEXT: br i1 [[CMP]], label %[[LOOP_1:.*]], label %[[EXIT:.*]] | |
| ; CHECK: [[LOOP_1]]: | |
| ; CHECK-NEXT: [[PTR_1:%.*]] = getelementptr inbounds i32, ptr [[P]], i32 1 | |
| ; CHECK-NEXT: store i32 0, ptr [[PTR_1]], align 4 | |
| ; CHECK-NEXT: br label %[[EXIT]] | |
| ; CHECK: [[EXIT]]: | |
| ; CHECK-NEXT: ret void | |
| ; | |
| entry: | |
| %mask = and i32 %n, 1 | |
| %bound = add nuw i32 %mask, 1 | |
| br label %loop | |
| loop: | |
| %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop ] | |
| %ptr = getelementptr inbounds i32, ptr %p, i32 %iv | |
| store i32 0, ptr %ptr, align 4 | |
| %iv.next = add nuw i32 %iv, 1 | |
| %cmp = icmp ult i32 %iv.next, %bound | |
| br i1 %cmp, label %loop, label %exit, !llvm.loop !0 | |
| exit: | |
| ret void | |
| } | |
| define void @full_only(ptr %p, i32 %n) { | |
| ; CHECK-LABEL: define void @full_only( | |
| ; CHECK-SAME: ptr [[P:%.*]], i32 [[N:%.*]]) { | |
| ; CHECK-NEXT: [[ENTRY:.*]]: | |
| ; CHECK-NEXT: [[MASK:%.*]] = and i32 [[N]], 1 | |
| ; CHECK-NEXT: [[BOUND:%.*]] = add nuw i32 [[MASK]], 1 | |
| ; CHECK-NEXT: br label %[[LOOP:.*]] | |
| ; CHECK: [[LOOP]]: | |
| ; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP]] ] | |
| ; CHECK-NEXT: [[PTR:%.*]] = getelementptr inbounds i32, ptr [[P]], i32 [[IV]] | |
| ; CHECK-NEXT: store i32 0, ptr [[PTR]], align 4 | |
| ; CHECK-NEXT: [[IV_NEXT]] = add nuw i32 [[IV]], 1 | |
| ; CHECK-NEXT: [[CMP:%.*]] = icmp ult i32 [[IV_NEXT]], [[BOUND]] | |
| ; CHECK-NEXT: br i1 [[CMP]], label %[[LOOP]], label %[[EXIT:.*]], !llvm.loop [[LOOP0:![0-9]+]] | |
| ; CHECK: [[EXIT]]: | |
| ; CHECK-NEXT: ret void | |
| ; | |
| entry: | |
| %mask = and i32 %n, 1 | |
| %bound = add nuw i32 %mask, 1 | |
| br label %loop | |
| loop: | |
| %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop ] | |
| %ptr = getelementptr inbounds i32, ptr %p, i32 %iv | |
| store i32 0, ptr %ptr, align 4 | |
| %iv.next = add nuw i32 %iv, 1 | |
| %cmp = icmp ult i32 %iv.next, %bound | |
| br i1 %cmp, label %loop, label %exit, !llvm.loop !1 | |
| exit: | |
| ret void | |
| } | |
| !0 = distinct !{!0, !2, !3} | |
| !1 = distinct !{!1, !2} | |
| !2 = !{!"llvm.loop.unroll.full"} | |
| !3 = !{!"llvm.loop.unroll.enable"} | |
| ;. | |
| ; CHECK: [[LOOP0]] = distinct !{[[LOOP0]], [[META1:![0-9]+]]} | |
| ; CHECK: [[META1]] = !{!"llvm.loop.unroll.full"} | |
| ;. |