diff options
-rw-r--r-- | lib/Target/README.txt | 4 | ||||
-rw-r--r-- | lib/Transforms/InstCombine/InstCombineShifts.cpp | 20 | ||||
-rw-r--r-- | test/Transforms/InstCombine/shift.ll | 6 |
3 files changed, 14 insertions, 16 deletions
diff --git a/lib/Target/README.txt b/lib/Target/README.txt index b98531c..38f5af6 100644 --- a/lib/Target/README.txt +++ b/lib/Target/README.txt @@ -1704,8 +1704,8 @@ Missed instcombine transformation: %384 = shl i64 %381, %383 ; [#uses=1] %385 = icmp slt i32 %tmp14.i, 64 ; [#uses=1] -The srem can be transformed to an and because if x is negative, the shift is -undefined. Testcase derived from 403.gcc. +The srem can be transformed to an and because if %tmp14.i is negative, the +shift is undefined. Testcase derived from 403.gcc. //===---------------------------------------------------------------------===// diff --git a/lib/Transforms/InstCombine/InstCombineShifts.cpp b/lib/Transforms/InstCombine/InstCombineShifts.cpp index 7969dbf..e52f201 100644 --- a/lib/Transforms/InstCombine/InstCombineShifts.cpp +++ b/lib/Transforms/InstCombine/InstCombineShifts.cpp @@ -54,19 +54,17 @@ Instruction *InstCombiner::commonShiftTransforms(BinaryOperator &I) { if (Instruction *Res = FoldShiftByConstant(Op0, CUI, I)) return Res; - // X shift (A srem B) -> X shift (A urem B) iff B is positive. + // X shift (A srem B) -> X shift (A and B-1) iff B is a power of 2. // Because shifts by negative values are undefined. if (BinaryOperator *BO = dyn_cast<BinaryOperator>(Op1)) - if (BO->getOpcode() == Instruction::SRem && BO->getType()->isIntegerTy()) { - // Make sure the divisor's sign bit is zero. - APInt Mask = APInt::getSignBit(BO->getType()->getPrimitiveSizeInBits()); - if (MaskedValueIsZero(BO->getOperand(1), Mask)) { - Value *URem = Builder->CreateURem(BO->getOperand(0), BO->getOperand(1), - BO->getName()); - I.setOperand(1, URem); - return &I; - } - } + if (BO->hasOneUse() && BO->getOpcode() == Instruction::SRem) + if (ConstantInt *CI = dyn_cast<ConstantInt>(BO->getOperand(1))) + if (CI->getValue().isPowerOf2()) { + Constant *C = ConstantInt::get(BO->getType(), CI->getValue()-1); + Value *Rem = Builder->CreateAnd(BO->getOperand(0), C, BO->getName()); + I.setOperand(1, Rem); + return &I; + } return 0; } diff --git a/test/Transforms/InstCombine/shift.ll b/test/Transforms/InstCombine/shift.ll index 6bebca9..8d1c829 100644 --- a/test/Transforms/InstCombine/shift.ll +++ b/test/Transforms/InstCombine/shift.ll @@ -443,12 +443,12 @@ entry: } define i32 @test38(i32 %x) nounwind readnone { -entry: %rem = srem i32 %x, 32 %shl = shl i32 1, %rem ret i32 %shl ; CHECK: @test38 -; CHECK-NOT: srem -; CHECK: ret i32 +; CHECK-NEXT: and i32 %x, 31 +; CHECK-NEXT: shl i32 1 +; CHECK-NEXT: ret i32 } |