summaryrefslogtreecommitdiffstats
path: root/llvm/lib
diff options
context:
space:
mode:
Diffstat (limited to 'llvm/lib')
-rw-r--r--llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp19
1 files changed, 19 insertions, 0 deletions
diff --git a/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp b/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
index 7f152aac52d..a301d9eef60 100644
--- a/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
+++ b/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
@@ -213,6 +213,25 @@ Instruction *InstCombiner::visitMul(BinaryOperator &I) {
if (Instruction *FoldedMul = foldBinOpIntoSelectOrPhi(I))
return FoldedMul;
+ // TODO: This is a specific form of a much more general pattern.
+ // We could detect a select with any binop identity constant, or we
+ // could use SimplifyBinOp to see if either arm of the select reduces.
+ // But that needs to be done carefully and/or while removing potential
+ // reverse canonicalizations as in InstCombiner::foldSelectIntoOp().
+ // mul (select Cond, 1, -1), Op1 --> select Cond, Op1, -Op1
+ // mul (select Cond, -1, 1), Op1 --> select Cond, -Op1, Op1
+ // mul Op0, (select Cond, 1, -1) --> select Cond, Op0, -Op0
+ // mul Op0, (select Cond, -1, 1) --> select Cond, -Op0, Op0
+ Value *Cond;
+ if (match(Op0, m_OneUse(m_Select(m_Value(Cond), m_One(), m_AllOnes()))))
+ return SelectInst::Create(Cond, Op1, Builder.CreateNeg(Op1));
+ if (match(Op0, m_OneUse(m_Select(m_Value(Cond), m_AllOnes(), m_One()))))
+ return SelectInst::Create(Cond, Builder.CreateNeg(Op1), Op1);
+ if (match(Op1, m_OneUse(m_Select(m_Value(Cond), m_One(), m_AllOnes()))))
+ return SelectInst::Create(Cond, Op0, Builder.CreateNeg(Op0));
+ if (match(Op1, m_OneUse(m_Select(m_Value(Cond), m_AllOnes(), m_One()))))
+ return SelectInst::Create(Cond, Builder.CreateNeg(Op0), Op0);
+
// Simplify mul instructions with a constant RHS.
if (isa<Constant>(Op1)) {
// Canonicalize (X+C1)*CI -> X*CI+C1*CI.
OpenPOWER on IntegriCloud