; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -passes=constraint-elimination -S %s | FileCheck %s


declare void @fn()

define i1 @test_pr54228(i32 %a, i32 %b, i1 %i.0, i1 %i.1) {
; CHECK-LABEL: @test_pr54228(
; CHECK-NEXT:  entry:
; CHECK-NEXT:    br i1 [[I_0:%.*]], label [[PH_1:%.*]], label [[LOOP_HEADER:%.*]]
; CHECK:       ph.1:
; CHECK-NEXT:    [[C_1:%.*]] = icmp eq i32 [[A:%.*]], 0
; CHECK-NEXT:    call void @llvm.assume(i1 [[C_1]])
; CHECK-NEXT:    br label [[LOOP_HEADER]]
; CHECK:       loop.header:
; CHECK-NEXT:    br i1 [[I_1:%.*]], label [[LOOP_THEN:%.*]], label [[LOOP_LATCH:%.*]]
; CHECK:       loop.then:
; CHECK-NEXT:    call void @fn()
; CHECK-NEXT:    [[C_2:%.*]] = icmp eq i32 [[B:%.*]], 0
; CHECK-NEXT:    call void @llvm.assume(i1 [[C_2]])
; CHECK-NEXT:    br label [[LOOP_LATCH]]
; CHECK:       loop.latch:
; CHECK-NEXT:    [[C_3:%.*]] = icmp eq i32 [[B]], 1
; CHECK-NEXT:    br i1 [[C_3]], label [[EXIT:%.*]], label [[LOOP_HEADER]]
; CHECK:       exit:
; CHECK-NEXT:    [[C_4:%.*]] = icmp eq i32 [[A]], 0
; CHECK-NEXT:    ret i1 [[C_4]]
;
entry:
  br i1 %i.0, label %ph.1, label %loop.header

ph.1:                                             ; preds = %entry
  %c.1 = icmp eq i32 %a, 0
  call void @llvm.assume(i1 %c.1)
  br label %loop.header

loop.header:                                      ; preds = %loop.latch, %ph.1, %entry
  br i1 %i.1, label %loop.then, label %loop.latch

loop.then:                                        ; preds = %loop.header
  call void @fn()
  %c.2 = icmp eq i32 %b, 0
  call void @llvm.assume(i1 %c.2)
  br label %loop.latch

loop.latch:                                       ; preds = %loop.then, %loop.header
  %c.3 = icmp eq i32 %b, 1
  br i1 %c.3, label %exit, label %loop.header

exit:                                             ; preds = %loop.latch
  %c.4 = icmp eq i32 %a, 0
  ret i1 %c.4
}

declare void @llvm.assume(i1 noundef)