[LVI] Support no constant range of cast and freeze value in getEdgeValueLocal.

Author: andjo403

llvm/lib/Analysis/LazyValueInfo.cpp

@@ -1493,6 +1493,24 @@ LazyValueInfoImpl::getEdgeValueLocal(Value *Val, BasicBlock *BBFrom,
             //   br %Condition, label %then, label %else
             APInt ConditionVal(1, isTrueDest ? 1 : 0);
             Result = constantFoldUser(Usr, Condition, ConditionVal, DL);
+          } else if (isa<TruncInst, ZExtInst, SExtInst, FreezeInst>(Usr)) {
+            ValueLatticeElement OpLatticeVal =
+                *getValueFromCondition(Usr->getOperand(0), Condition,
+                                       isTrueDest, /*UseBlockValue*/ false);
+
+            if (isa<FreezeInst>(Usr) || !OpLatticeVal.isConstantRange())
+              return OpLatticeVal;
+
+            const unsigned ResultBitWidth =
+                Usr->getType()->getScalarSizeInBits();
+            if (auto *Trunc = dyn_cast<TruncInst>(Usr))
+              return ValueLatticeElement::getRange(
+                  OpLatticeVal.getConstantRange().truncate(
+                      ResultBitWidth, Trunc->getNoWrapKind()));
+
+            return ValueLatticeElement::getRange(
+                OpLatticeVal.getConstantRange().castOp(
+                    cast<CastInst>(Usr)->getOpcode(), ResultBitWidth));
           } else {
             // If one of Val's operand has an inferred value, we may be able to
             // infer the value of Val.

llvm/test/Transforms/CorrelatedValuePropagation/range.ll

@@ -1236,7 +1236,7 @@ define i1 @neg_icmp_eq_range_call() {
 }
 
 define i16 @return_range_for_edge_value_zext(i8 %a) {
-; CHECK-LABEL: define range(i16 0, 256) i16 @return_range_for_edge_value_zext(
+; CHECK-LABEL: define range(i16 0, 98) i16 @return_range_for_edge_value_zext(
 ; CHECK-SAME: i8 [[A:%.*]]) {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[B:%.*]] = zext i8 [[A]] to i16
@@ -1260,7 +1260,7 @@ target:
 }
 
 define i16 @return_range_for_edge_value_sext(i8 %a) {
-; CHECK-LABEL: define range(i16 -128, 128) i16 @return_range_for_edge_value_sext(
+; CHECK-LABEL: define range(i16 0, 98) i16 @return_range_for_edge_value_sext(
 ; CHECK-SAME: i8 [[A:%.*]]) {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[B:%.*]] = sext i8 [[A]] to i16
@@ -1284,7 +1284,7 @@ target:
 }
 
 define i8 @return_range_for_edge_value_trunc(i16 %a) {
-; CHECK-LABEL: define i8 @return_range_for_edge_value_trunc(
+; CHECK-LABEL: define range(i8 0, 98) i8 @return_range_for_edge_value_trunc(
 ; CHECK-SAME: i16 [[A:%.*]]) {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[B:%.*]] = trunc i16 [[A]] to i8
@@ -1332,7 +1332,7 @@ target:
 }
 
 define i8 @return_range_for_edge_value_trunc_nuw(i16 %a) {
-; CHECK-LABEL: define i8 @return_range_for_edge_value_trunc_nuw(
+; CHECK-LABEL: define range(i8 -55, 0) i8 @return_range_for_edge_value_trunc_nuw(
 ; CHECK-SAME: i16 [[A:%.*]]) {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[B:%.*]] = trunc nuw i16 [[A]] to i8
@@ -1356,7 +1356,7 @@ target:
 }
 
 define i8 @return_range_for_edge_value_freeze(i8 %a) {
-; CHECK-LABEL: define i8 @return_range_for_edge_value_freeze(
+; CHECK-LABEL: define range(i8 0, 98) i8 @return_range_for_edge_value_freeze(
 ; CHECK-SAME: i8 [[A:%.*]]) {
 ; CHECK-NEXT:  [[ENTRY:.*:]]
 ; CHECK-NEXT:    [[B:%.*]] = freeze i8 [[A]]

llvm/test/Transforms/JumpThreading/branch-debug-info2.ll

@@ -19,7 +19,7 @@ if.end.i60:                                       ; preds = %shadow_to_ptr.exit
 ; Jump threading is going to fold the select in to the branch. Ensure debug
 ; info is not lost, and is merged from the select and the branch.
 ; CHECK-NOT: br i1 %cmp3.i, label %for.inc, label %ptr_to_shadow.exit
-; CHECK: br i1 %cmp3.i, label %for.inc, label %ptr_to_shadow.exit, !dbg [[DBG:![0-9]+]]
+; CHECK: br i1 %cmp3.i, label %for.inc, label %if.end22, !dbg [[DBG:![0-9]+]]
 ; CHECK: [[DBG]] = !DILocation(line: 9, column: 1, scope: !{{.*}})
 
   br label %ptr_to_shadow.exit, !dbg !17