; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals all --version 5
; RUN: opt -p loop-unroll -scev-cheap-expansion-budget=2 -S %s | FileCheck --check-prefix=BUDGET2 %s
; RUN: opt -p loop-unroll -scev-cheap-expansion-budget=3 -S %s | FileCheck --check-prefix=BUDGET3 %s

; Note that BUDGET3 will expose a conditional branch going to the same label.
; This is expected, and the profile information associated will be dropped when
; the branch is simplified to unconditional.

define i32 @test_expansion_cost_2(i32 %start, i32 %end) !prof !0 {
; BUDGET2-LABEL: define i32 @test_expansion_cost_2(
; BUDGET2-SAME: i32 [[START:%.*]], i32 [[END:%.*]]) !prof [[PROF0:![0-9]+]] {
; BUDGET2-NEXT:  [[ENTRY:.*]]:
; BUDGET2-NEXT:    [[SUB:%.*]] = add i32 [[END]], -1
; BUDGET2-NEXT:    br label %[[LOOP_HEADER:.*]]
; BUDGET2:       [[LOOP_HEADER]]:
; BUDGET2-NEXT:    [[IV:%.*]] = phi i32 [ [[START]], %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
; BUDGET2-NEXT:    [[C:%.*]] = icmp eq i32 [[IV]], [[SUB]]
; BUDGET2-NEXT:    br i1 [[C]], label %[[THEN:.*]], label %[[LOOP_LATCH]], !prof [[PROF1:![0-9]+]]
; BUDGET2:       [[THEN]]:
; BUDGET2-NEXT:    br label %[[LOOP_LATCH]]
; BUDGET2:       [[LOOP_LATCH]]:
; BUDGET2-NEXT:    [[IV_NEXT]] = add nsw i32 [[IV]], 1
; BUDGET2-NEXT:    [[EC:%.*]] = icmp eq i32 [[IV_NEXT]], [[END]]
; BUDGET2-NEXT:    br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP_HEADER]], !prof [[PROF2:![0-9]+]], !llvm.loop [[LOOP3:![0-9]+]]
; BUDGET2:       [[EXIT]]:
; BUDGET2-NEXT:    ret i32 0
;
; BUDGET3-LABEL: define i32 @test_expansion_cost_2(
; BUDGET3-SAME: i32 [[START:%.*]], i32 [[END:%.*]]) !prof [[PROF0:![0-9]+]] {
; BUDGET3-NEXT:  [[ENTRY:.*]]:
; BUDGET3-NEXT:    [[SUB:%.*]] = add i32 [[END]], -1
; BUDGET3-NEXT:    [[TMP0:%.*]] = sub i32 [[SUB]], [[START]]
; BUDGET3-NEXT:    [[TMP1:%.*]] = icmp ne i32 [[TMP0]], 0
; BUDGET3-NEXT:    br i1 [[TMP1]], label %[[ENTRY_SPLIT:.*]], label %[[EXIT_PEEL_BEGIN:.*]], !prof [[PROF1:![0-9]+]]
; BUDGET3:       [[ENTRY_SPLIT]]:
; BUDGET3-NEXT:    br label %[[LOOP_HEADER:.*]]
; BUDGET3:       [[LOOP_HEADER]]:
; BUDGET3-NEXT:    [[IV:%.*]] = phi i32 [ [[START]], %[[ENTRY_SPLIT]] ], [ [[IV_NEXT:%.*]], %[[LOOP_LATCH:.*]] ]
; BUDGET3-NEXT:    [[C:%.*]] = icmp eq i32 [[IV]], [[SUB]]
; BUDGET3-NEXT:    br i1 [[C]], label %[[THEN:.*]], label %[[LOOP_LATCH]], !prof [[PROF2:![0-9]+]]
; BUDGET3:       [[THEN]]:
; BUDGET3-NEXT:    br label %[[LOOP_LATCH]]
; BUDGET3:       [[LOOP_LATCH]]:
; BUDGET3-NEXT:    [[IV_NEXT]] = add nsw i32 [[IV]], 1
; BUDGET3-NEXT:    [[TMP2:%.*]] = sub i32 [[END]], 1
; BUDGET3-NEXT:    [[EC:%.*]] = icmp eq i32 [[IV_NEXT]], [[TMP2]]
; BUDGET3-NEXT:    br i1 [[EC]], label %[[EXIT_PEEL_BEGIN_LOOPEXIT:.*]], label %[[LOOP_HEADER]], !prof [[PROF3:![0-9]+]], !llvm.loop [[LOOP4:![0-9]+]]
; BUDGET3:       [[EXIT_PEEL_BEGIN_LOOPEXIT]]:
; BUDGET3-NEXT:    [[DOTPH:%.*]] = phi i32 [ [[IV_NEXT]], %[[LOOP_LATCH]] ]
; BUDGET3-NEXT:    br label %[[EXIT_PEEL_BEGIN]]
; BUDGET3:       [[EXIT_PEEL_BEGIN]]:
; BUDGET3-NEXT:    [[TMP3:%.*]] = phi i32 [ [[START]], %[[ENTRY]] ], [ [[DOTPH]], %[[EXIT_PEEL_BEGIN_LOOPEXIT]] ]
; BUDGET3-NEXT:    br label %[[LOOP_HEADER_PEEL:.*]]
; BUDGET3:       [[LOOP_HEADER_PEEL]]:
; BUDGET3-NEXT:    [[C_PEEL:%.*]] = icmp eq i32 [[TMP3]], [[SUB]]
; BUDGET3-NEXT:    br i1 [[C_PEEL]], label %[[THEN_PEEL:.*]], label %[[LOOP_LATCH_PEEL:.*]], !prof [[PROF2]]
; BUDGET3:       [[THEN_PEEL]]:
; BUDGET3-NEXT:    br label %[[LOOP_LATCH_PEEL]]
; BUDGET3:       [[LOOP_LATCH_PEEL]]:
; BUDGET3-NEXT:    [[IV_NEXT_PEEL:%.*]] = add nsw i32 [[TMP3]], 1
; BUDGET3-NEXT:    [[EC_PEEL:%.*]] = icmp eq i32 [[IV_NEXT_PEEL]], [[END]]
; BUDGET3-NEXT:    br i1 [[EC_PEEL]], label %[[EXIT_PEEL_NEXT:.*]], label %[[EXIT_PEEL_NEXT]], !prof [[PROF3]]
; BUDGET3:       [[EXIT_PEEL_NEXT]]:
; BUDGET3-NEXT:    br label %[[LOOP_HEADER_PEEL_NEXT:.*]]
; BUDGET3:       [[LOOP_HEADER_PEEL_NEXT]]:
; BUDGET3-NEXT:    br label %[[EXIT:.*]]
; BUDGET3:       [[EXIT]]:
; BUDGET3-NEXT:    ret i32 0
;
entry:
  %sub = add i32 %end, -1
  br label %loop.header

loop.header:
  %iv = phi i32 [ %start, %entry ], [ %iv.next, %loop.latch ]
  %c = icmp eq i32 %iv, %sub
  br i1 %c, label %then, label %loop.latch, !prof !1

then:
  br label %loop.latch

loop.latch:
  %iv.next = add nsw i32 %iv, 1
  %ec = icmp eq i32 %iv.next, %end
  br i1 %ec, label %exit, label %loop.header, !prof !2, !llvm.loop !3

exit:
  ret i32 0
}

!0 = !{!"function_entry_count", i32 10}
!1 = !{!"branch_weights", i32 2, i32 3}
!2 = !{!"branch_weights", i32 1, i32 10}
!3 = distinct !{!3, !4}
!4 = !{!"llvm.loop.estimated_trip_count", i32 42}
;.
; BUDGET2: [[PROF0]] = !{!"function_entry_count", i32 10}
; BUDGET2: [[PROF1]] = !{!"branch_weights", i32 2, i32 3}
; BUDGET2: [[PROF2]] = !{!"branch_weights", i32 1, i32 10}
; BUDGET2: [[LOOP3]] = distinct !{[[LOOP3]], [[META4:![0-9]+]]}
; BUDGET2: [[META4]] = !{!"llvm.loop.estimated_trip_count", i32 42}
;.
; BUDGET3: [[PROF0]] = !{!"function_entry_count", i32 10}
; BUDGET3: [[PROF1]] = !{!"branch_weights", i32 10, i32 1}
; BUDGET3: [[PROF2]] = !{!"branch_weights", i32 2, i32 3}
; BUDGET3: [[PROF3]] = !{!"branch_weights", i32 1, i32 10}
; BUDGET3: [[LOOP4]] = distinct !{[[LOOP4]], [[META5:![0-9]+]], [[META6:![0-9]+]]}
; BUDGET3: [[META5]] = !{!"llvm.loop.peeled.count", i32 1}
; BUDGET3: [[META6]] = !{!"llvm.loop.estimated_trip_count", i32 41}
;.