; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals
; RUN: opt < %s -S -passes=loop-unroll -unroll-force-peel-count=2 2>&1 | FileCheck %s
; RUN: opt < %s -S -passes=loop-unroll -unroll-force-peel-count=2 -disable-advanced-peeling 2>&1 | FileCheck %s --check-prefix=DISABLEADV

declare i32 @get.x()

; Test branch weight update for terminator with multiple fallthrough and
; multiple exit edges.
define void @test() {
; CHECK-LABEL: @test(
; CHECK-NEXT:  entry:
; CHECK-NEXT:    br label [[LOOP_PEEL_BEGIN:%.*]]
; CHECK:       loop.peel.begin:
; CHECK-NEXT:    br label [[LOOP_PEEL:%.*]]
; CHECK:       loop.peel:
; CHECK-NEXT:    [[X_PEEL:%.*]] = call i32 @get.x()
; CHECK-NEXT:    switch i32 [[X_PEEL]], label [[LOOP_LATCH_PEEL:%.*]] [
; CHECK-NEXT:      i32 0, label [[LOOP_LATCH_PEEL]]
; CHECK-NEXT:      i32 1, label [[LOOP_EXIT:%.*]]
; CHECK-NEXT:      i32 2, label [[LOOP_EXIT]]
; CHECK-NEXT:    ], !prof [[PROF0:![0-9]+]]
; CHECK:       loop.latch.peel:
; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT:%.*]]
; CHECK:       loop.peel.next:
; CHECK-NEXT:    br label [[LOOP_PEEL2:%.*]]
; CHECK:       loop.peel2:
; CHECK-NEXT:    [[X_PEEL3:%.*]] = call i32 @get.x()
; CHECK-NEXT:    switch i32 [[X_PEEL3]], label [[LOOP_LATCH_PEEL4:%.*]] [
; CHECK-NEXT:      i32 0, label [[LOOP_LATCH_PEEL4]]
; CHECK-NEXT:      i32 1, label [[LOOP_EXIT]]
; CHECK-NEXT:      i32 2, label [[LOOP_EXIT]]
; CHECK-NEXT:    ], !prof [[PROF0]]
; CHECK:       loop.latch.peel4:
; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT1:%.*]]
; CHECK:       loop.peel.next1:
; CHECK-NEXT:    br label [[LOOP_PEEL_NEXT5:%.*]]
; CHECK:       loop.peel.next5:
; CHECK-NEXT:    br label [[ENTRY_PEEL_NEWPH:%.*]]
; CHECK:       entry.peel.newph:
; CHECK-NEXT:    br label [[LOOP:%.*]]
; CHECK:       loop:
; CHECK-NEXT:    [[X:%.*]] = call i32 @get.x()
; CHECK-NEXT:    switch i32 [[X]], label [[LOOP_LATCH:%.*]] [
; CHECK-NEXT:      i32 0, label [[LOOP_LATCH]]
; CHECK-NEXT:      i32 1, label [[LOOP_EXIT_LOOPEXIT:%.*]]
; CHECK-NEXT:      i32 2, label [[LOOP_EXIT_LOOPEXIT]]
; CHECK-NEXT:    ], !prof [[PROF0]]
; CHECK:       loop.latch:
; CHECK-NEXT:    br label [[LOOP]], !llvm.loop [[LOOP1:![0-9]+]]
; CHECK:       loop.exit.loopexit:
; CHECK-NEXT:    br label [[LOOP_EXIT]]
; CHECK:       loop.exit:
; CHECK-NEXT:    ret void
;
; DISABLEADV-LABEL: @test(
; DISABLEADV-NEXT:  entry:
; DISABLEADV-NEXT:    br label [[LOOP:%.*]]
; DISABLEADV:       loop:
; DISABLEADV-NEXT:    [[X:%.*]] = call i32 @get.x()
; DISABLEADV-NEXT:    switch i32 [[X]], label [[LOOP_LATCH:%.*]] [
; DISABLEADV-NEXT:      i32 0, label [[LOOP_LATCH]]
; DISABLEADV-NEXT:      i32 1, label [[LOOP_EXIT:%.*]]
; DISABLEADV-NEXT:      i32 2, label [[LOOP_EXIT]]
; DISABLEADV-NEXT:    ], !prof [[PROF0:![0-9]+]]
; DISABLEADV:       loop.latch:
; DISABLEADV-NEXT:    br label [[LOOP]]
; DISABLEADV:       loop.exit:
; DISABLEADV-NEXT:    ret void
;


entry:
  br label %loop

loop:
  %x = call i32 @get.x()
  switch i32 %x, label %loop.latch [
  i32 0, label %loop.latch
  i32 1, label %loop.exit
  i32 2, label %loop.exit
  ], !prof !0

loop.latch:
  br label %loop

loop.exit:
  ret void
}

!0 = !{!"branch_weights", i32 100, i32 200, i32 20, i32 10}

;.
; CHECK: [[PROF0]] = !{!"branch_weights", i32 100, i32 200, i32 20, i32 10}
; CHECK: [[LOOP1]] = distinct !{[[LOOP1]], [[META2:![0-9]+]], [[META3:![0-9]+]]}
; CHECK: [[META2]] = !{!"llvm.loop.peeled.count", i32 2}
; CHECK: [[META3]] = !{!"llvm.loop.unroll.disable"}
;.
; DISABLEADV: [[PROF0]] = !{!"branch_weights", i32 100, i32 200, i32 20, i32 10}
;.