; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -passes=correlated-propagation -S %s | FileCheck %s
; Check that shift is not removed by CVP because of incorrect range returned by LVI::getConstantRangeAtUse.
; https://github.com/llvm/llvm-project/issues/60629
define i32 @test(i32 %arg) {
; CHECK-LABEL: @test(
; CHECK-NEXT: entry:
; CHECK-NEXT: br label [[LOOP:%.*]]
; CHECK: loop:
; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 127, [[ENTRY:%.*]] ], [ [[IV_NEXT:%.*]], [[DO_SHIFT:%.*]] ]
; CHECK-NEXT: [[SHIFTED_IV:%.*]] = phi i32 [ 0, [[ENTRY]] ], [ [[SHIFT:%.*]], [[DO_SHIFT]] ]
; CHECK-NEXT: [[IV_NEXT]] = add i32 [[IV]], 1
; CHECK-NEXT: [[LOOP_COND:%.*]] = icmp sgt i32 [[IV]], 127
; CHECK-NEXT: br i1 [[LOOP_COND]], label [[EXIT:%.*]], label [[DO_SHIFT]]
; CHECK: do.shift:
; CHECK-NEXT: [[SHIFT]] = ashr i32 [[IV]], [[ARG:%.*]]
; CHECK-NEXT: br label [[LOOP]]
; CHECK: bb1:
; CHECK-NEXT: br label [[EXIT]]
; CHECK: exit:
; CHECK-NEXT: [[RETVAL:%.*]] = phi i32 [ 0, [[BB1:%.*]] ], [ [[SHIFTED_IV]], [[LOOP]] ]
; CHECK-NEXT: ret i32 [[RETVAL]]
;
entry:
br label %loop
loop: ; preds = %do.shift, %entry
%iv = phi i32 [ 127, %entry ], [ %iv.next, %do.shift ]
%shifted.iv = phi i32 [ 0, %entry ], [ %shift, %do.shift ]
%iv.next = add i32 %iv, 1
%loop_cond = icmp sgt i32 %iv, 127
br i1 %loop_cond, label %exit, label %do.shift
do.shift: ; preds = %loop
%shift = ashr i32 %iv, %arg
br label %loop
bb1: ; No predecessors!
br label %exit
exit: ; preds = %loop, %bb1
%retval = phi i32 [ 0, %bb1 ], [ %shifted.iv, %loop ]
ret i32 %retval
}