[ConstraintElim] Don't decompose values wider than 64 bits (llvm#68803)

Our coefficients are 64-bits, so adding/multiplying them can wrap in
64-bits even if there would be no wrapping the full bit width.

The alternative would be to check for overflows during all adds/muls in
decomposition. I assume that we don't particularly care about handling
wide integers here, so I've opted to bail out.

Fixes llvm#68751.

(cherry picked from commit 1d43096)

llvm/lib/Transforms/Scalar/ConstraintElimination.cpp

@@ -412,6 +412,19 @@ static Decomposition decompose(Value *V,
  return ResA;
  };
 
+ Type *Ty = V->getType()->getScalarType();
+ if (Ty->isPointerTy() && !IsSigned) {
+ if (auto *GEP = dyn_cast<GEPOperator>(V))
+ return decomposeGEP(*GEP, Preconditions, IsSigned, DL);
+ return V;
+ }
+
+ // Don't handle integers > 64 bit. Our coefficients are 64-bit large, so
+ // coefficient add/mul may wrap, while the operation in the full bit width
+ // would not.
+ if (!Ty->isIntegerTy() || Ty->getIntegerBitWidth() > 64)
+ return V;
+
  // Decompose \p V used with a signed predicate.
  if (IsSigned) {
  if (auto *CI = dyn_cast<ConstantInt>(V)) {
@@ -439,9 +452,6 @@ static Decomposition decompose(Value *V,
  return int64_t(CI->getZExtValue());
  }
 
- if (auto *GEP = dyn_cast<GEPOperator>(V))
- return decomposeGEP(*GEP, Preconditions, IsSigned, DL);
-
  Value *Op0;
  bool IsKnownNonNegative = false;
  if (match(V, m_ZExt(m_Value(Op0)))) {

llvm/test/Transforms/ConstraintElimination/large-constant-ints.ll

@@ -96,6 +96,7 @@ else:
  ret i1 false
 }
 
+; TODO: This could be folded.
 define i1 @sub_decomp_i80(i80 %a) {
 ; CHECK-LABEL: @sub_decomp_i80(
 ; CHECK-NEXT: entry:
@@ -105,7 +106,7 @@ define i1 @sub_decomp_i80(i80 %a) {
 ; CHECK: then:
 ; CHECK-NEXT: [[SUB_1:%.*]] = sub nuw i80 [[A]], 1973801615886922022913
 ; CHECK-NEXT: [[C_1:%.*]] = icmp ult i80 [[SUB_1]], 1346612317380797267967
-; CHECK-NEXT: ret i1 true
+; CHECK-NEXT: ret i1 [[C_1]]
 ; CHECK: else:
 ; CHECK-NEXT: ret i1 false
 ;
@@ -423,12 +424,12 @@ entry:
  ret i1 %res
 }
 
-; FIXME: This is a miscompile.
 define i1 @pr68751(i128 %arg) {
 ; CHECK-LABEL: @pr68751(
 ; CHECK-NEXT: [[SHL1:%.*]] = shl nuw nsw i128 [[ARG:%.*]], 32
 ; CHECK-NEXT: [[SHL2:%.*]] = shl nuw nsw i128 [[SHL1]], 32
-; CHECK-NEXT: ret i1 true
+; CHECK-NEXT: [[CMP:%.*]] = icmp eq i128 [[SHL2]], 0
+; CHECK-NEXT: ret i1 [[CMP]]
 ;
  %shl1 = shl nuw nsw i128 %arg, 32
  %shl2 = shl nuw nsw i128 %shl1, 32

llvm/test/Transforms/ConstraintElimination/shl.ll

@@ -1281,7 +1281,7 @@ define i1 @shl_55() {
 ; CHECK-NEXT: entry:
 ; CHECK-NEXT: [[SHL_UB:%.*]] = shl nuw nsw i256 1, 55
 ; CHECK-NEXT: [[SHL_CMP:%.*]] = icmp uge i256 [[SHL_UB]], 1
-; CHECK-NEXT: ret i1 true
+; CHECK-NEXT: ret i1 [[SHL_CMP]]
 ;
 entry:
  %shl.ub = shl nuw nsw i256 1, 55