Index: llvm/lib/IR/ConstantRange.cpp =================================================================== --- llvm/lib/IR/ConstantRange.cpp +++ llvm/lib/IR/ConstantRange.cpp @@ -256,6 +256,91 @@ APInt::getSignedMinValue(BitWidth) + SignedMin)); } return Result; + case Instruction::Mul: { + if (NoWrapKind == (OBO::NoSignedWrap | OBO::NoUnsignedWrap)) { + return SubsetIntersect( + makeGuaranteedNoWrapRegion(BinOp, Other, OBO::NoSignedWrap), + makeGuaranteedNoWrapRegion(BinOp, Other, OBO::NoUnsignedWrap)); + } + + // Equivalent to calling makeGuaranteedNoWrapRegion() on [V, V+1). + bool Unsigned = NoWrapKind == OBO::NoUnsignedWrap; + const auto makeSingleValueRegion = [Unsigned, + BitWidth](APInt V) -> ConstantRange { + const auto DivRoundDown = [Unsigned](APInt A, APInt B) -> APInt { + if (Unsigned) { + return A.udiv(B); + } + APInt Quo, Rem; + APInt::sdivrem(A, B, Quo, Rem); + if (Rem == 0) { + return Quo; + } + if (Rem.isNegative() != B.isNegative()) { + return Quo - 1; + } + return Quo; + }; + const auto DivRoundUp = [Unsigned](APInt A, APInt B) -> APInt { + if (Unsigned) { + APInt Quo, Rem; + APInt::udivrem(A, B, Quo, Rem); + if (Rem == 0) { + return Quo; + } + return Quo + 1; + } + APInt Quo, Rem; + APInt::sdivrem(A, B, Quo, Rem); + if (Rem == 0) { + return Quo; + } + if (Rem.isNegative() != B.isNegative()) { + return Quo; + } + return Quo + 1; + }; + + if (V == 0 || V.isOneValue()) { + return ConstantRange(BitWidth, true); + } + + APInt MinValue, MaxValue; + if (Unsigned) { + MinValue = APInt::getMinValue(BitWidth); + MaxValue = APInt::getMaxValue(BitWidth); + } else { + MinValue = APInt::getSignedMinValue(BitWidth); + MaxValue = APInt::getSignedMaxValue(BitWidth); + } + if (!Unsigned && V.isAllOnesValue()) { + // e.g. [-127, 127] -> [-127, -128). + return ConstantRange(-MaxValue, MinValue); + } + + APInt Lower, Upper; + if (!Unsigned && V.isNegative()) { + Lower = DivRoundUp(MaxValue, V); + Upper = DivRoundDown(MinValue, V); + } else { + Lower = DivRoundUp(MinValue, V); + Upper = DivRoundDown(MaxValue, V); + } + if (Unsigned) { + Lower = Lower.zextOrSelf(BitWidth); + Upper = Upper.zextOrSelf(BitWidth); + } else { + Lower = Lower.sextOrSelf(BitWidth); + Upper = Upper.sextOrSelf(BitWidth); + } + return ConstantRange(Lower, Upper + 1); + }; + return SubsetIntersect( + makeSingleValueRegion(Unsigned ? Other.getUnsignedMin() + : Other.getSignedMin()), + makeSingleValueRegion(Unsigned ? Other.getUnsignedMax() + : Other.getSignedMax())); + } } } Index: llvm/unittests/IR/ConstantRangeTest.cpp =================================================================== --- llvm/unittests/IR/ConstantRangeTest.cpp +++ llvm/unittests/IR/ConstantRangeTest.cpp @@ -1021,4 +1021,91 @@ EXPECT_EQ(RHS, APInt(32, -1)); } +TEST(ConstantRange, MakeGuaranteedNoWrapRegionMulSingleValue) { + typedef OverflowingBinaryOperator OBO; + + for (uint64_t i = std::numeric_limits::min(); + i <= std::numeric_limits::max(); i++) { + auto Range = ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, ConstantRange(APInt(8, i), APInt(8, i + 1)), + OBO::NoUnsignedWrap); + + for (uint64_t v = std::numeric_limits::min(); + v <= std::numeric_limits::max(); v++) { + bool Overflow; + (void)APInt(8, i).umul_ov(APInt(8, v), Overflow); + EXPECT_EQ(!Overflow, Range.contains(APInt(8, v))); + } + } + + for (int64_t i = std::numeric_limits::min(); + i <= std::numeric_limits::max(); i++) { + auto Range = ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, + ConstantRange(APInt(8, i, true), APInt(8, i + 1, true)), + OBO::NoSignedWrap); + + for (int64_t v = std::numeric_limits::min(); + v <= std::numeric_limits::max(); v++) { + bool Overflow; + (void)APInt(8, i, true).smul_ov(APInt(8, v, true), Overflow); + EXPECT_EQ(!Overflow, Range.contains(APInt(8, v, true))); + } + } + + for (uint64_t i = std::numeric_limits::min(); + i <= std::numeric_limits::max(); i++) { + auto Range = ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, ConstantRange(APInt(8, i), APInt(8, i + 1)), + OBO::NoUnsignedWrap | OBO::NoSignedWrap); + + for (uint64_t v = std::numeric_limits::min(); + v <= std::numeric_limits::max(); v++) { + bool UOverflow; + (void)APInt(8, i).umul_ov(APInt(8, v), UOverflow); + bool SOverflow; + (void)APInt(8, i).smul_ov(APInt(8, v), SOverflow); + EXPECT_EQ(!(UOverflow || SOverflow), Range.contains(APInt(8, v))); + } + } +} + +TEST(ConstantRange, MakeGuaranteedNoWrapRegionMulUnsignedRange) { + typedef OverflowingBinaryOperator OBO; + + for (uint64_t lo = std::numeric_limits::min(); + lo <= std::numeric_limits::max(); lo++) { + for (uint64_t hi = lo; hi <= std::numeric_limits::max(); hi++) { + EXPECT_EQ( + ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, ConstantRange(APInt(8, lo), APInt(8, hi + 1)), + OBO::NoUnsignedWrap), + ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, ConstantRange(APInt(8, hi), APInt(8, hi + 1)), + OBO::NoUnsignedWrap)); + } + } +} + +TEST(ConstantRange, MakeGuaranteedNoWrapRegionMulSignedRange) { + typedef OverflowingBinaryOperator OBO; + + int lo = -12, hi = 16; + auto Range = ConstantRange::makeGuaranteedNoWrapRegion( + Instruction::Mul, + ConstantRange(APInt(8, lo, true), APInt(8, hi + 1, true)), + OBO::NoSignedWrap); + + for (int64_t v = std::numeric_limits::min(); + v <= std::numeric_limits::max(); v++) { + bool AnyOverflow = false; + for (int64_t i = lo; i <= hi; i++) { + bool Overflow; + (void)APInt(8, i, true).smul_ov(APInt(8, v, true), Overflow); + AnyOverflow |= Overflow; + } + EXPECT_EQ(!AnyOverflow, Range.contains(APInt(8, v, true))); + } +} + } // anonymous namespace