Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 1 | //===- InstCombineAddSub.cpp ------------------------------------*- C++ -*-===// |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file implements the visit functions for add, fadd, sub, and fsub. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Chandler Carruth | a917458 | 2015-01-22 05:25:13 +0000 | [diff] [blame] | 14 | #include "InstCombineInternal.h" |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/APFloat.h" |
| 16 | #include "llvm/ADT/APInt.h" |
Craig Topper | 5871321 | 2013-07-15 04:27:47 +0000 | [diff] [blame] | 17 | #include "llvm/ADT/STLExtras.h" |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/SmallVector.h" |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 19 | #include "llvm/Analysis/InstructionSimplify.h" |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 20 | #include "llvm/Analysis/ValueTracking.h" |
| 21 | #include "llvm/IR/Constant.h" |
| 22 | #include "llvm/IR/Constants.h" |
| 23 | #include "llvm/IR/InstrTypes.h" |
| 24 | #include "llvm/IR/Instruction.h" |
| 25 | #include "llvm/IR/Instructions.h" |
| 26 | #include "llvm/IR/Operator.h" |
Chandler Carruth | 820a908 | 2014-03-04 11:08:18 +0000 | [diff] [blame] | 27 | #include "llvm/IR/PatternMatch.h" |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 28 | #include "llvm/IR/Type.h" |
| 29 | #include "llvm/IR/Value.h" |
| 30 | #include "llvm/Support/AlignOf.h" |
| 31 | #include "llvm/Support/Casting.h" |
Craig Topper | b45eabc | 2017-04-26 16:39:58 +0000 | [diff] [blame] | 32 | #include "llvm/Support/KnownBits.h" |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 33 | #include <cassert> |
| 34 | #include <utility> |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 35 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 36 | using namespace llvm; |
| 37 | using namespace PatternMatch; |
| 38 | |
Chandler Carruth | 964daaa | 2014-04-22 02:55:47 +0000 | [diff] [blame] | 39 | #define DEBUG_TYPE "instcombine" |
| 40 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 41 | namespace { |
| 42 | |
| 43 | /// Class representing coefficient of floating-point addend. |
| 44 | /// This class needs to be highly efficient, which is especially true for |
| 45 | /// the constructor. As of I write this comment, the cost of the default |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 46 | /// constructor is merely 4-byte-store-zero (Assuming compiler is able to |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 47 | /// perform write-merging). |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 48 | /// |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 49 | class FAddendCoef { |
| 50 | public: |
Suyog Sarda | de409fd | 2014-07-17 06:09:34 +0000 | [diff] [blame] | 51 | // The constructor has to initialize a APFloat, which is unnecessary for |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 52 | // most addends which have coefficient either 1 or -1. So, the constructor |
| 53 | // is expensive. In order to avoid the cost of the constructor, we should |
| 54 | // reuse some instances whenever possible. The pre-created instances |
| 55 | // FAddCombine::Add[0-5] embodies this idea. |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 56 | FAddendCoef() = default; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 57 | ~FAddendCoef(); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 58 | |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 59 | // If possible, don't define operator+/operator- etc because these |
| 60 | // operators inevitably call FAddendCoef's constructor which is not cheap. |
| 61 | void operator=(const FAddendCoef &A); |
| 62 | void operator+=(const FAddendCoef &A); |
| 63 | void operator*=(const FAddendCoef &S); |
| 64 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 65 | void set(short C) { |
| 66 | assert(!insaneIntVal(C) && "Insane coefficient"); |
| 67 | IsFp = false; IntVal = C; |
| 68 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 69 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 70 | void set(const APFloat& C); |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 71 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 72 | void negate(); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 73 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 74 | bool isZero() const { return isInt() ? !IntVal : getFpVal().isZero(); } |
| 75 | Value *getValue(Type *) const; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 76 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 77 | bool isOne() const { return isInt() && IntVal == 1; } |
| 78 | bool isTwo() const { return isInt() && IntVal == 2; } |
| 79 | bool isMinusOne() const { return isInt() && IntVal == -1; } |
| 80 | bool isMinusTwo() const { return isInt() && IntVal == -2; } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 81 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 82 | private: |
| 83 | bool insaneIntVal(int V) { return V > 4 || V < -4; } |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 84 | |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 85 | APFloat *getFpValPtr() |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 86 | { return reinterpret_cast<APFloat *>(&FpValBuf.buffer[0]); } |
| 87 | |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 88 | const APFloat *getFpValPtr() const |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 89 | { return reinterpret_cast<const APFloat *>(&FpValBuf.buffer[0]); } |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 90 | |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 91 | const APFloat &getFpVal() const { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 92 | assert(IsFp && BufHasFpVal && "Incorret state"); |
David Greene | 530430b | 2013-01-14 21:04:40 +0000 | [diff] [blame] | 93 | return *getFpValPtr(); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 94 | } |
| 95 | |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 96 | APFloat &getFpVal() { |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 97 | assert(IsFp && BufHasFpVal && "Incorret state"); |
| 98 | return *getFpValPtr(); |
| 99 | } |
| 100 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 101 | bool isInt() const { return !IsFp; } |
| 102 | |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 103 | // If the coefficient is represented by an integer, promote it to a |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 104 | // floating point. |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 105 | void convertToFpType(const fltSemantics &Sem); |
| 106 | |
| 107 | // Construct an APFloat from a signed integer. |
| 108 | // TODO: We should get rid of this function when APFloat can be constructed |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 109 | // from an *SIGNED* integer. |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 110 | APFloat createAPFloatFromInt(const fltSemantics &Sem, int Val); |
Shuxin Yang | 5b841c4 | 2012-12-19 01:10:17 +0000 | [diff] [blame] | 111 | |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 112 | bool IsFp = false; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 113 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 114 | // True iff FpValBuf contains an instance of APFloat. |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 115 | bool BufHasFpVal = false; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 116 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 117 | // The integer coefficient of an individual addend is either 1 or -1, |
| 118 | // and we try to simplify at most 4 addends from neighboring at most |
| 119 | // two instructions. So the range of <IntVal> falls in [-4, 4]. APInt |
| 120 | // is overkill of this end. |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 121 | short IntVal = 0; |
Shuxin Yang | 5b841c4 | 2012-12-19 01:10:17 +0000 | [diff] [blame] | 122 | |
| 123 | AlignedCharArrayUnion<APFloat> FpValBuf; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 124 | }; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 125 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 126 | /// FAddend is used to represent floating-point addend. An addend is |
| 127 | /// represented as <C, V>, where the V is a symbolic value, and C is a |
| 128 | /// constant coefficient. A constant addend is represented as <C, 0>. |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 129 | class FAddend { |
| 130 | public: |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 131 | FAddend() = default; |
| 132 | |
| 133 | void operator+=(const FAddend &T) { |
| 134 | assert((Val == T.Val) && "Symbolic-values disagree"); |
| 135 | Coeff += T.Coeff; |
| 136 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 137 | |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 138 | Value *getSymVal() const { return Val; } |
| 139 | const FAddendCoef &getCoef() const { return Coeff; } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 140 | |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 141 | bool isConstant() const { return Val == nullptr; } |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 142 | bool isZero() const { return Coeff.isZero(); } |
| 143 | |
Richard Trieu | 7a08381 | 2016-02-18 22:09:30 +0000 | [diff] [blame] | 144 | void set(short Coefficient, Value *V) { |
| 145 | Coeff.set(Coefficient); |
| 146 | Val = V; |
| 147 | } |
| 148 | void set(const APFloat &Coefficient, Value *V) { |
| 149 | Coeff.set(Coefficient); |
| 150 | Val = V; |
| 151 | } |
| 152 | void set(const ConstantFP *Coefficient, Value *V) { |
| 153 | Coeff.set(Coefficient->getValueAPF()); |
| 154 | Val = V; |
| 155 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 156 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 157 | void negate() { Coeff.negate(); } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 158 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 159 | /// Drill down the U-D chain one step to find the definition of V, and |
| 160 | /// try to break the definition into one or two addends. |
| 161 | static unsigned drillValueDownOneStep(Value* V, FAddend &A0, FAddend &A1); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 162 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 163 | /// Similar to FAddend::drillDownOneStep() except that the value being |
| 164 | /// splitted is the addend itself. |
| 165 | unsigned drillAddendDownOneStep(FAddend &Addend0, FAddend &Addend1) const; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 166 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 167 | private: |
| 168 | void Scale(const FAddendCoef& ScaleAmt) { Coeff *= ScaleAmt; } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 169 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 170 | // This addend has the value of "Coeff * Val". |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 171 | Value *Val = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 172 | FAddendCoef Coeff; |
| 173 | }; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 174 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 175 | /// FAddCombine is the class for optimizing an unsafe fadd/fsub along |
| 176 | /// with its neighboring at most two instructions. |
| 177 | /// |
| 178 | class FAddCombine { |
| 179 | public: |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 180 | FAddCombine(InstCombiner::BuilderTy &B) : Builder(B) {} |
| 181 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 182 | Value *simplify(Instruction *FAdd); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 183 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 184 | private: |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 185 | using AddendVect = SmallVector<const FAddend *, 4>; |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 186 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 187 | Value *simplifyFAdd(AddendVect& V, unsigned InstrQuota); |
Shuxin Yang | 2eca602 | 2013-03-14 18:08:26 +0000 | [diff] [blame] | 188 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 189 | /// Convert given addend to a Value |
| 190 | Value *createAddendVal(const FAddend &A, bool& NeedNeg); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 191 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 192 | /// Return the number of instructions needed to emit the N-ary addition. |
| 193 | unsigned calcInstrNumber(const AddendVect& Vect); |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 194 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 195 | Value *createFSub(Value *Opnd0, Value *Opnd1); |
| 196 | Value *createFAdd(Value *Opnd0, Value *Opnd1); |
| 197 | Value *createFMul(Value *Opnd0, Value *Opnd1); |
| 198 | Value *createFNeg(Value *V); |
| 199 | Value *createNaryFAdd(const AddendVect& Opnds, unsigned InstrQuota); |
Owen Anderson | 1664dc8 | 2014-01-20 07:44:53 +0000 | [diff] [blame] | 200 | void createInstPostProc(Instruction *NewInst, bool NoNumber = false); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 201 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 202 | // Debugging stuff are clustered here. |
| 203 | #ifndef NDEBUG |
| 204 | unsigned CreateInstrNum; |
| 205 | void initCreateInstNum() { CreateInstrNum = 0; } |
| 206 | void incCreateInstNum() { CreateInstrNum++; } |
| 207 | #else |
| 208 | void initCreateInstNum() {} |
| 209 | void incCreateInstNum() {} |
| 210 | #endif |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 211 | |
| 212 | InstCombiner::BuilderTy &Builder; |
| 213 | Instruction *Instr = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 214 | }; |
Eugene Zelenko | ffec81c | 2015-11-04 22:32:32 +0000 | [diff] [blame] | 215 | |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 216 | } // end anonymous namespace |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 217 | |
| 218 | //===----------------------------------------------------------------------===// |
| 219 | // |
| 220 | // Implementation of |
| 221 | // {FAddendCoef, FAddend, FAddition, FAddCombine}. |
| 222 | // |
| 223 | //===----------------------------------------------------------------------===// |
| 224 | FAddendCoef::~FAddendCoef() { |
| 225 | if (BufHasFpVal) |
| 226 | getFpValPtr()->~APFloat(); |
| 227 | } |
| 228 | |
| 229 | void FAddendCoef::set(const APFloat& C) { |
| 230 | APFloat *P = getFpValPtr(); |
| 231 | |
| 232 | if (isInt()) { |
| 233 | // As the buffer is meanless byte stream, we cannot call |
| 234 | // APFloat::operator=(). |
| 235 | new(P) APFloat(C); |
| 236 | } else |
| 237 | *P = C; |
| 238 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 239 | IsFp = BufHasFpVal = true; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 240 | } |
| 241 | |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 242 | void FAddendCoef::convertToFpType(const fltSemantics &Sem) { |
| 243 | if (!isInt()) |
| 244 | return; |
| 245 | |
| 246 | APFloat *P = getFpValPtr(); |
| 247 | if (IntVal > 0) |
| 248 | new(P) APFloat(Sem, IntVal); |
| 249 | else { |
| 250 | new(P) APFloat(Sem, 0 - IntVal); |
| 251 | P->changeSign(); |
| 252 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 253 | IsFp = BufHasFpVal = true; |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 254 | } |
| 255 | |
| 256 | APFloat FAddendCoef::createAPFloatFromInt(const fltSemantics &Sem, int Val) { |
| 257 | if (Val >= 0) |
| 258 | return APFloat(Sem, Val); |
| 259 | |
| 260 | APFloat T(Sem, 0 - Val); |
| 261 | T.changeSign(); |
| 262 | |
| 263 | return T; |
| 264 | } |
| 265 | |
| 266 | void FAddendCoef::operator=(const FAddendCoef &That) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 267 | if (That.isInt()) |
| 268 | set(That.IntVal); |
| 269 | else |
| 270 | set(That.getFpVal()); |
| 271 | } |
| 272 | |
| 273 | void FAddendCoef::operator+=(const FAddendCoef &That) { |
| 274 | enum APFloat::roundingMode RndMode = APFloat::rmNearestTiesToEven; |
| 275 | if (isInt() == That.isInt()) { |
| 276 | if (isInt()) |
| 277 | IntVal += That.IntVal; |
| 278 | else |
| 279 | getFpVal().add(That.getFpVal(), RndMode); |
| 280 | return; |
| 281 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 282 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 283 | if (isInt()) { |
| 284 | const APFloat &T = That.getFpVal(); |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 285 | convertToFpType(T.getSemantics()); |
| 286 | getFpVal().add(T, RndMode); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 287 | return; |
| 288 | } |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 289 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 290 | APFloat &T = getFpVal(); |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 291 | T.add(createAPFloatFromInt(T.getSemantics(), That.IntVal), RndMode); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 292 | } |
| 293 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 294 | void FAddendCoef::operator*=(const FAddendCoef &That) { |
| 295 | if (That.isOne()) |
| 296 | return; |
| 297 | |
| 298 | if (That.isMinusOne()) { |
| 299 | negate(); |
| 300 | return; |
| 301 | } |
| 302 | |
| 303 | if (isInt() && That.isInt()) { |
| 304 | int Res = IntVal * (int)That.IntVal; |
| 305 | assert(!insaneIntVal(Res) && "Insane int value"); |
| 306 | IntVal = Res; |
| 307 | return; |
| 308 | } |
| 309 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 310 | const fltSemantics &Semantic = |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 311 | isInt() ? That.getFpVal().getSemantics() : getFpVal().getSemantics(); |
| 312 | |
| 313 | if (isInt()) |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 314 | convertToFpType(Semantic); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 315 | APFloat &F0 = getFpVal(); |
| 316 | |
| 317 | if (That.isInt()) |
Shuxin Yang | 389ed4b | 2013-03-25 20:43:41 +0000 | [diff] [blame] | 318 | F0.multiply(createAPFloatFromInt(Semantic, That.IntVal), |
| 319 | APFloat::rmNearestTiesToEven); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 320 | else |
| 321 | F0.multiply(That.getFpVal(), APFloat::rmNearestTiesToEven); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 322 | } |
| 323 | |
| 324 | void FAddendCoef::negate() { |
| 325 | if (isInt()) |
| 326 | IntVal = 0 - IntVal; |
| 327 | else |
| 328 | getFpVal().changeSign(); |
| 329 | } |
| 330 | |
| 331 | Value *FAddendCoef::getValue(Type *Ty) const { |
| 332 | return isInt() ? |
| 333 | ConstantFP::get(Ty, float(IntVal)) : |
| 334 | ConstantFP::get(Ty->getContext(), getFpVal()); |
| 335 | } |
| 336 | |
| 337 | // The definition of <Val> Addends |
| 338 | // ========================================= |
| 339 | // A + B <1, A>, <1,B> |
| 340 | // A - B <1, A>, <1,B> |
| 341 | // 0 - B <-1, B> |
| 342 | // C * A, <C, A> |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 343 | // A + C <1, A> <C, NULL> |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 344 | // 0 +/- 0 <0, NULL> (corner case) |
| 345 | // |
| 346 | // Legend: A and B are not constant, C is constant |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 347 | unsigned FAddend::drillValueDownOneStep |
| 348 | (Value *Val, FAddend &Addend0, FAddend &Addend1) { |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 349 | Instruction *I = nullptr; |
| 350 | if (!Val || !(I = dyn_cast<Instruction>(Val))) |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 351 | return 0; |
| 352 | |
| 353 | unsigned Opcode = I->getOpcode(); |
| 354 | |
| 355 | if (Opcode == Instruction::FAdd || Opcode == Instruction::FSub) { |
| 356 | ConstantFP *C0, *C1; |
| 357 | Value *Opnd0 = I->getOperand(0); |
| 358 | Value *Opnd1 = I->getOperand(1); |
| 359 | if ((C0 = dyn_cast<ConstantFP>(Opnd0)) && C0->isZero()) |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 360 | Opnd0 = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 361 | |
| 362 | if ((C1 = dyn_cast<ConstantFP>(Opnd1)) && C1->isZero()) |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 363 | Opnd1 = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 364 | |
| 365 | if (Opnd0) { |
| 366 | if (!C0) |
| 367 | Addend0.set(1, Opnd0); |
| 368 | else |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 369 | Addend0.set(C0, nullptr); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 370 | } |
| 371 | |
| 372 | if (Opnd1) { |
| 373 | FAddend &Addend = Opnd0 ? Addend1 : Addend0; |
| 374 | if (!C1) |
| 375 | Addend.set(1, Opnd1); |
| 376 | else |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 377 | Addend.set(C1, nullptr); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 378 | if (Opcode == Instruction::FSub) |
| 379 | Addend.negate(); |
| 380 | } |
| 381 | |
| 382 | if (Opnd0 || Opnd1) |
| 383 | return Opnd0 && Opnd1 ? 2 : 1; |
| 384 | |
| 385 | // Both operands are zero. Weird! |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 386 | Addend0.set(APFloat(C0->getValueAPF().getSemantics()), nullptr); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 387 | return 1; |
| 388 | } |
| 389 | |
| 390 | if (I->getOpcode() == Instruction::FMul) { |
| 391 | Value *V0 = I->getOperand(0); |
| 392 | Value *V1 = I->getOperand(1); |
| 393 | if (ConstantFP *C = dyn_cast<ConstantFP>(V0)) { |
| 394 | Addend0.set(C, V1); |
| 395 | return 1; |
| 396 | } |
| 397 | |
| 398 | if (ConstantFP *C = dyn_cast<ConstantFP>(V1)) { |
| 399 | Addend0.set(C, V0); |
| 400 | return 1; |
| 401 | } |
| 402 | } |
| 403 | |
| 404 | return 0; |
| 405 | } |
| 406 | |
| 407 | // Try to break *this* addend into two addends. e.g. Suppose this addend is |
| 408 | // <2.3, V>, and V = X + Y, by calling this function, we obtain two addends, |
| 409 | // i.e. <2.3, X> and <2.3, Y>. |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 410 | unsigned FAddend::drillAddendDownOneStep |
| 411 | (FAddend &Addend0, FAddend &Addend1) const { |
| 412 | if (isConstant()) |
| 413 | return 0; |
| 414 | |
| 415 | unsigned BreakNum = FAddend::drillValueDownOneStep(Val, Addend0, Addend1); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 416 | if (!BreakNum || Coeff.isOne()) |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 417 | return BreakNum; |
| 418 | |
| 419 | Addend0.Scale(Coeff); |
| 420 | |
| 421 | if (BreakNum == 2) |
| 422 | Addend1.Scale(Coeff); |
| 423 | |
| 424 | return BreakNum; |
| 425 | } |
| 426 | |
| 427 | Value *FAddCombine::simplify(Instruction *I) { |
Warren Ristow | 8b2f27c | 2018-04-14 19:18:28 +0000 | [diff] [blame] | 428 | assert(I->hasAllowReassoc() && I->hasNoSignedZeros() && |
| 429 | "Expected 'reassoc'+'nsz' instruction"); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 430 | |
| 431 | // Currently we are not able to handle vector type. |
| 432 | if (I->getType()->isVectorTy()) |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 433 | return nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 434 | |
| 435 | assert((I->getOpcode() == Instruction::FAdd || |
| 436 | I->getOpcode() == Instruction::FSub) && "Expect add/sub"); |
| 437 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 438 | // Save the instruction before calling other member-functions. |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 439 | Instr = I; |
| 440 | |
| 441 | FAddend Opnd0, Opnd1, Opnd0_0, Opnd0_1, Opnd1_0, Opnd1_1; |
| 442 | |
| 443 | unsigned OpndNum = FAddend::drillValueDownOneStep(I, Opnd0, Opnd1); |
| 444 | |
| 445 | // Step 1: Expand the 1st addend into Opnd0_0 and Opnd0_1. |
| 446 | unsigned Opnd0_ExpNum = 0; |
| 447 | unsigned Opnd1_ExpNum = 0; |
| 448 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 449 | if (!Opnd0.isConstant()) |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 450 | Opnd0_ExpNum = Opnd0.drillAddendDownOneStep(Opnd0_0, Opnd0_1); |
| 451 | |
| 452 | // Step 2: Expand the 2nd addend into Opnd1_0 and Opnd1_1. |
| 453 | if (OpndNum == 2 && !Opnd1.isConstant()) |
| 454 | Opnd1_ExpNum = Opnd1.drillAddendDownOneStep(Opnd1_0, Opnd1_1); |
| 455 | |
| 456 | // Step 3: Try to optimize Opnd0_0 + Opnd0_1 + Opnd1_0 + Opnd1_1 |
| 457 | if (Opnd0_ExpNum && Opnd1_ExpNum) { |
| 458 | AddendVect AllOpnds; |
| 459 | AllOpnds.push_back(&Opnd0_0); |
| 460 | AllOpnds.push_back(&Opnd1_0); |
| 461 | if (Opnd0_ExpNum == 2) |
| 462 | AllOpnds.push_back(&Opnd0_1); |
| 463 | if (Opnd1_ExpNum == 2) |
| 464 | AllOpnds.push_back(&Opnd1_1); |
| 465 | |
| 466 | // Compute instruction quota. We should save at least one instruction. |
| 467 | unsigned InstQuota = 0; |
| 468 | |
| 469 | Value *V0 = I->getOperand(0); |
| 470 | Value *V1 = I->getOperand(1); |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 471 | InstQuota = ((!isa<Constant>(V0) && V0->hasOneUse()) && |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 472 | (!isa<Constant>(V1) && V1->hasOneUse())) ? 2 : 1; |
| 473 | |
| 474 | if (Value *R = simplifyFAdd(AllOpnds, InstQuota)) |
| 475 | return R; |
| 476 | } |
| 477 | |
| 478 | if (OpndNum != 2) { |
| 479 | // The input instruction is : "I=0.0 +/- V". If the "V" were able to be |
| 480 | // splitted into two addends, say "V = X - Y", the instruction would have |
| 481 | // been optimized into "I = Y - X" in the previous steps. |
| 482 | // |
| 483 | const FAddendCoef &CE = Opnd0.getCoef(); |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 484 | return CE.isOne() ? Opnd0.getSymVal() : nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 485 | } |
| 486 | |
| 487 | // step 4: Try to optimize Opnd0 + Opnd1_0 [+ Opnd1_1] |
| 488 | if (Opnd1_ExpNum) { |
| 489 | AddendVect AllOpnds; |
| 490 | AllOpnds.push_back(&Opnd0); |
| 491 | AllOpnds.push_back(&Opnd1_0); |
| 492 | if (Opnd1_ExpNum == 2) |
| 493 | AllOpnds.push_back(&Opnd1_1); |
| 494 | |
| 495 | if (Value *R = simplifyFAdd(AllOpnds, 1)) |
| 496 | return R; |
| 497 | } |
| 498 | |
| 499 | // step 5: Try to optimize Opnd1 + Opnd0_0 [+ Opnd0_1] |
| 500 | if (Opnd0_ExpNum) { |
| 501 | AddendVect AllOpnds; |
| 502 | AllOpnds.push_back(&Opnd1); |
| 503 | AllOpnds.push_back(&Opnd0_0); |
| 504 | if (Opnd0_ExpNum == 2) |
| 505 | AllOpnds.push_back(&Opnd0_1); |
| 506 | |
| 507 | if (Value *R = simplifyFAdd(AllOpnds, 1)) |
| 508 | return R; |
| 509 | } |
| 510 | |
Sanjay Patel | dc185ee | 2018-08-12 15:48:26 +0000 | [diff] [blame] | 511 | return nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 512 | } |
| 513 | |
| 514 | Value *FAddCombine::simplifyFAdd(AddendVect& Addends, unsigned InstrQuota) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 515 | unsigned AddendNum = Addends.size(); |
| 516 | assert(AddendNum <= 4 && "Too many addends"); |
| 517 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 518 | // For saving intermediate results; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 519 | unsigned NextTmpIdx = 0; |
| 520 | FAddend TmpResult[3]; |
| 521 | |
| 522 | // Points to the constant addend of the resulting simplified expression. |
| 523 | // If the resulting expr has constant-addend, this constant-addend is |
| 524 | // desirable to reside at the top of the resulting expression tree. Placing |
| 525 | // constant close to supper-expr(s) will potentially reveal some optimization |
| 526 | // opportunities in super-expr(s). |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 527 | const FAddend *ConstAdd = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 528 | |
| 529 | // Simplified addends are placed <SimpVect>. |
| 530 | AddendVect SimpVect; |
| 531 | |
| 532 | // The outer loop works on one symbolic-value at a time. Suppose the input |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 533 | // addends are : <a1, x>, <b1, y>, <a2, x>, <c1, z>, <b2, y>, ... |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 534 | // The symbolic-values will be processed in this order: x, y, z. |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 535 | for (unsigned SymIdx = 0; SymIdx < AddendNum; SymIdx++) { |
| 536 | |
| 537 | const FAddend *ThisAddend = Addends[SymIdx]; |
| 538 | if (!ThisAddend) { |
| 539 | // This addend was processed before. |
| 540 | continue; |
| 541 | } |
| 542 | |
| 543 | Value *Val = ThisAddend->getSymVal(); |
| 544 | unsigned StartIdx = SimpVect.size(); |
| 545 | SimpVect.push_back(ThisAddend); |
| 546 | |
| 547 | // The inner loop collects addends sharing same symbolic-value, and these |
| 548 | // addends will be later on folded into a single addend. Following above |
| 549 | // example, if the symbolic value "y" is being processed, the inner loop |
| 550 | // will collect two addends "<b1,y>" and "<b2,Y>". These two addends will |
| 551 | // be later on folded into "<b1+b2, y>". |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 552 | for (unsigned SameSymIdx = SymIdx + 1; |
| 553 | SameSymIdx < AddendNum; SameSymIdx++) { |
| 554 | const FAddend *T = Addends[SameSymIdx]; |
| 555 | if (T && T->getSymVal() == Val) { |
| 556 | // Set null such that next iteration of the outer loop will not process |
| 557 | // this addend again. |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 558 | Addends[SameSymIdx] = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 559 | SimpVect.push_back(T); |
| 560 | } |
| 561 | } |
| 562 | |
| 563 | // If multiple addends share same symbolic value, fold them together. |
| 564 | if (StartIdx + 1 != SimpVect.size()) { |
| 565 | FAddend &R = TmpResult[NextTmpIdx ++]; |
| 566 | R = *SimpVect[StartIdx]; |
| 567 | for (unsigned Idx = StartIdx + 1; Idx < SimpVect.size(); Idx++) |
| 568 | R += *SimpVect[Idx]; |
| 569 | |
| 570 | // Pop all addends being folded and push the resulting folded addend. |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 571 | SimpVect.resize(StartIdx); |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 572 | if (Val) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 573 | if (!R.isZero()) { |
| 574 | SimpVect.push_back(&R); |
| 575 | } |
| 576 | } else { |
| 577 | // Don't push constant addend at this time. It will be the last element |
| 578 | // of <SimpVect>. |
| 579 | ConstAdd = &R; |
| 580 | } |
| 581 | } |
| 582 | } |
| 583 | |
Craig Topper | 5871321 | 2013-07-15 04:27:47 +0000 | [diff] [blame] | 584 | assert((NextTmpIdx <= array_lengthof(TmpResult) + 1) && |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 585 | "out-of-bound access"); |
| 586 | |
| 587 | if (ConstAdd) |
| 588 | SimpVect.push_back(ConstAdd); |
| 589 | |
| 590 | Value *Result; |
| 591 | if (!SimpVect.empty()) |
| 592 | Result = createNaryFAdd(SimpVect, InstrQuota); |
| 593 | else { |
| 594 | // The addition is folded to 0.0. |
| 595 | Result = ConstantFP::get(Instr->getType(), 0.0); |
| 596 | } |
| 597 | |
| 598 | return Result; |
| 599 | } |
| 600 | |
| 601 | Value *FAddCombine::createNaryFAdd |
| 602 | (const AddendVect &Opnds, unsigned InstrQuota) { |
| 603 | assert(!Opnds.empty() && "Expect at least one addend"); |
| 604 | |
| 605 | // Step 1: Check if the # of instructions needed exceeds the quota. |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 606 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 607 | unsigned InstrNeeded = calcInstrNumber(Opnds); |
| 608 | if (InstrNeeded > InstrQuota) |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 609 | return nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 610 | |
| 611 | initCreateInstNum(); |
| 612 | |
| 613 | // step 2: Emit the N-ary addition. |
| 614 | // Note that at most three instructions are involved in Fadd-InstCombine: the |
| 615 | // addition in question, and at most two neighboring instructions. |
| 616 | // The resulting optimized addition should have at least one less instruction |
| 617 | // than the original addition expression tree. This implies that the resulting |
| 618 | // N-ary addition has at most two instructions, and we don't need to worry |
| 619 | // about tree-height when constructing the N-ary addition. |
| 620 | |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 621 | Value *LastVal = nullptr; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 622 | bool LastValNeedNeg = false; |
| 623 | |
| 624 | // Iterate the addends, creating fadd/fsub using adjacent two addends. |
Benjamin Kramer | 135f735 | 2016-06-26 12:28:59 +0000 | [diff] [blame] | 625 | for (const FAddend *Opnd : Opnds) { |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 626 | bool NeedNeg; |
Benjamin Kramer | 135f735 | 2016-06-26 12:28:59 +0000 | [diff] [blame] | 627 | Value *V = createAddendVal(*Opnd, NeedNeg); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 628 | if (!LastVal) { |
| 629 | LastVal = V; |
| 630 | LastValNeedNeg = NeedNeg; |
| 631 | continue; |
| 632 | } |
| 633 | |
| 634 | if (LastValNeedNeg == NeedNeg) { |
| 635 | LastVal = createFAdd(LastVal, V); |
| 636 | continue; |
| 637 | } |
| 638 | |
| 639 | if (LastValNeedNeg) |
| 640 | LastVal = createFSub(V, LastVal); |
| 641 | else |
| 642 | LastVal = createFSub(LastVal, V); |
| 643 | |
| 644 | LastValNeedNeg = false; |
| 645 | } |
| 646 | |
| 647 | if (LastValNeedNeg) { |
| 648 | LastVal = createFNeg(LastVal); |
| 649 | } |
| 650 | |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 651 | #ifndef NDEBUG |
| 652 | assert(CreateInstrNum == InstrNeeded && |
| 653 | "Inconsistent in instruction numbers"); |
| 654 | #endif |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 655 | |
| 656 | return LastVal; |
| 657 | } |
| 658 | |
Sanjay Patel | c242dbb | 2014-12-18 21:11:09 +0000 | [diff] [blame] | 659 | Value *FAddCombine::createFSub(Value *Opnd0, Value *Opnd1) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 660 | Value *V = Builder.CreateFSub(Opnd0, Opnd1); |
Shuxin Yang | 2eca602 | 2013-03-14 18:08:26 +0000 | [diff] [blame] | 661 | if (Instruction *I = dyn_cast<Instruction>(V)) |
| 662 | createInstPostProc(I); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 663 | return V; |
| 664 | } |
| 665 | |
| 666 | Value *FAddCombine::createFNeg(Value *V) { |
Sanjay Patel | ea3c802 | 2014-12-19 16:44:08 +0000 | [diff] [blame] | 667 | Value *Zero = cast<Value>(ConstantFP::getZeroValueForNegation(V->getType())); |
Owen Anderson | 1664dc8 | 2014-01-20 07:44:53 +0000 | [diff] [blame] | 668 | Value *NewV = createFSub(Zero, V); |
| 669 | if (Instruction *I = dyn_cast<Instruction>(NewV)) |
| 670 | createInstPostProc(I, true); // fneg's don't receive instruction numbers. |
| 671 | return NewV; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 672 | } |
| 673 | |
Sanjay Patel | c242dbb | 2014-12-18 21:11:09 +0000 | [diff] [blame] | 674 | Value *FAddCombine::createFAdd(Value *Opnd0, Value *Opnd1) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 675 | Value *V = Builder.CreateFAdd(Opnd0, Opnd1); |
Shuxin Yang | 2eca602 | 2013-03-14 18:08:26 +0000 | [diff] [blame] | 676 | if (Instruction *I = dyn_cast<Instruction>(V)) |
| 677 | createInstPostProc(I); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 678 | return V; |
| 679 | } |
| 680 | |
| 681 | Value *FAddCombine::createFMul(Value *Opnd0, Value *Opnd1) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 682 | Value *V = Builder.CreateFMul(Opnd0, Opnd1); |
Shuxin Yang | 2eca602 | 2013-03-14 18:08:26 +0000 | [diff] [blame] | 683 | if (Instruction *I = dyn_cast<Instruction>(V)) |
| 684 | createInstPostProc(I); |
| 685 | return V; |
| 686 | } |
| 687 | |
Sanjay Patel | c242dbb | 2014-12-18 21:11:09 +0000 | [diff] [blame] | 688 | void FAddCombine::createInstPostProc(Instruction *NewInstr, bool NoNumber) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 689 | NewInstr->setDebugLoc(Instr->getDebugLoc()); |
| 690 | |
| 691 | // Keep track of the number of instruction created. |
Owen Anderson | 1664dc8 | 2014-01-20 07:44:53 +0000 | [diff] [blame] | 692 | if (!NoNumber) |
| 693 | incCreateInstNum(); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 694 | |
| 695 | // Propagate fast-math flags |
| 696 | NewInstr->setFastMathFlags(Instr->getFastMathFlags()); |
| 697 | } |
| 698 | |
| 699 | // Return the number of instruction needed to emit the N-ary addition. |
| 700 | // NOTE: Keep this function in sync with createAddendVal(). |
| 701 | unsigned FAddCombine::calcInstrNumber(const AddendVect &Opnds) { |
| 702 | unsigned OpndNum = Opnds.size(); |
| 703 | unsigned InstrNeeded = OpndNum - 1; |
| 704 | |
Jim Grosbach | bdbd734 | 2013-04-05 21:20:12 +0000 | [diff] [blame] | 705 | // The number of addends in the form of "(-1)*x". |
| 706 | unsigned NegOpndNum = 0; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 707 | |
| 708 | // Adjust the number of instructions needed to emit the N-ary add. |
Benjamin Kramer | 135f735 | 2016-06-26 12:28:59 +0000 | [diff] [blame] | 709 | for (const FAddend *Opnd : Opnds) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 710 | if (Opnd->isConstant()) |
| 711 | continue; |
| 712 | |
Matt Arsenault | 02907f3 | 2017-04-24 17:24:37 +0000 | [diff] [blame] | 713 | // The constant check above is really for a few special constant |
| 714 | // coefficients. |
| 715 | if (isa<UndefValue>(Opnd->getSymVal())) |
| 716 | continue; |
| 717 | |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 718 | const FAddendCoef &CE = Opnd->getCoef(); |
| 719 | if (CE.isMinusOne() || CE.isMinusTwo()) |
| 720 | NegOpndNum++; |
| 721 | |
| 722 | // Let the addend be "c * x". If "c == +/-1", the value of the addend |
| 723 | // is immediately available; otherwise, it needs exactly one instruction |
| 724 | // to evaluate the value. |
| 725 | if (!CE.isMinusOne() && !CE.isOne()) |
| 726 | InstrNeeded++; |
| 727 | } |
| 728 | if (NegOpndNum == OpndNum) |
| 729 | InstrNeeded++; |
| 730 | return InstrNeeded; |
| 731 | } |
| 732 | |
| 733 | // Input Addend Value NeedNeg(output) |
| 734 | // ================================================================ |
| 735 | // Constant C C false |
| 736 | // <+/-1, V> V coefficient is -1 |
| 737 | // <2/-2, V> "fadd V, V" coefficient is -2 |
| 738 | // <C, V> "fmul V, C" false |
| 739 | // |
| 740 | // NOTE: Keep this function in sync with FAddCombine::calcInstrNumber. |
Sanjay Patel | c242dbb | 2014-12-18 21:11:09 +0000 | [diff] [blame] | 741 | Value *FAddCombine::createAddendVal(const FAddend &Opnd, bool &NeedNeg) { |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 742 | const FAddendCoef &Coeff = Opnd.getCoef(); |
| 743 | |
| 744 | if (Opnd.isConstant()) { |
| 745 | NeedNeg = false; |
| 746 | return Coeff.getValue(Instr->getType()); |
| 747 | } |
| 748 | |
| 749 | Value *OpndVal = Opnd.getSymVal(); |
| 750 | |
| 751 | if (Coeff.isMinusOne() || Coeff.isOne()) { |
| 752 | NeedNeg = Coeff.isMinusOne(); |
| 753 | return OpndVal; |
| 754 | } |
| 755 | |
| 756 | if (Coeff.isTwo() || Coeff.isMinusTwo()) { |
| 757 | NeedNeg = Coeff.isMinusTwo(); |
| 758 | return createFAdd(OpndVal, OpndVal); |
| 759 | } |
| 760 | |
| 761 | NeedNeg = false; |
| 762 | return createFMul(OpndVal, Coeff.getValue(Instr->getType())); |
| 763 | } |
| 764 | |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 765 | // Checks if any operand is negative and we can convert add to sub. |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 766 | // This function checks for following negative patterns |
| 767 | // ADD(XOR(OR(Z, NOT(C)), C)), 1) == NEG(AND(Z, C)) |
| 768 | // ADD(XOR(AND(Z, C), C), 1) == NEG(OR(Z, ~C)) |
| 769 | // XOR(AND(Z, C), (C + 1)) == NEG(OR(Z, ~C)) if C is even |
Benjamin Kramer | 6cbe670 | 2014-07-07 14:47:51 +0000 | [diff] [blame] | 770 | static Value *checkForNegativeOperand(BinaryOperator &I, |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 771 | InstCombiner::BuilderTy &Builder) { |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 772 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 773 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 774 | // This function creates 2 instructions to replace ADD, we need at least one |
| 775 | // of LHS or RHS to have one use to ensure benefit in transform. |
| 776 | if (!LHS->hasOneUse() && !RHS->hasOneUse()) |
| 777 | return nullptr; |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 778 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 779 | Value *X = nullptr, *Y = nullptr, *Z = nullptr; |
| 780 | const APInt *C1 = nullptr, *C2 = nullptr; |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 781 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 782 | // if ONE is on other side, swap |
| 783 | if (match(RHS, m_Add(m_Value(X), m_One()))) |
| 784 | std::swap(LHS, RHS); |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 785 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 786 | if (match(LHS, m_Add(m_Value(X), m_One()))) { |
| 787 | // if XOR on other side, swap |
| 788 | if (match(RHS, m_Xor(m_Value(Y), m_APInt(C1)))) |
| 789 | std::swap(X, RHS); |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 790 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 791 | if (match(X, m_Xor(m_Value(Y), m_APInt(C1)))) { |
| 792 | // X = XOR(Y, C1), Y = OR(Z, C2), C2 = NOT(C1) ==> X == NOT(AND(Z, C1)) |
| 793 | // ADD(ADD(X, 1), RHS) == ADD(X, ADD(RHS, 1)) == SUB(RHS, AND(Z, C1)) |
| 794 | if (match(Y, m_Or(m_Value(Z), m_APInt(C2))) && (*C2 == ~(*C1))) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 795 | Value *NewAnd = Builder.CreateAnd(Z, *C1); |
| 796 | return Builder.CreateSub(RHS, NewAnd, "sub"); |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 797 | } else if (match(Y, m_And(m_Value(Z), m_APInt(C2))) && (*C1 == *C2)) { |
| 798 | // X = XOR(Y, C1), Y = AND(Z, C2), C2 == C1 ==> X == NOT(OR(Z, ~C1)) |
| 799 | // ADD(ADD(X, 1), RHS) == ADD(X, ADD(RHS, 1)) == SUB(RHS, OR(Z, ~C1)) |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 800 | Value *NewOr = Builder.CreateOr(Z, ~(*C1)); |
| 801 | return Builder.CreateSub(RHS, NewOr, "sub"); |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 802 | } |
| 803 | } |
| 804 | } |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 805 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 806 | // Restore LHS and RHS |
| 807 | LHS = I.getOperand(0); |
| 808 | RHS = I.getOperand(1); |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 809 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 810 | // if XOR is on other side, swap |
| 811 | if (match(RHS, m_Xor(m_Value(Y), m_APInt(C1)))) |
| 812 | std::swap(LHS, RHS); |
| 813 | |
| 814 | // C2 is ODD |
| 815 | // LHS = XOR(Y, C1), Y = AND(Z, C2), C1 == (C2 + 1) => LHS == NEG(OR(Z, ~C2)) |
| 816 | // ADD(LHS, RHS) == SUB(RHS, OR(Z, ~C2)) |
| 817 | if (match(LHS, m_Xor(m_Value(Y), m_APInt(C1)))) |
| 818 | if (C1->countTrailingZeros() == 0) |
| 819 | if (match(Y, m_And(m_Value(Z), m_APInt(C2))) && *C1 == (*C2 + 1)) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 820 | Value *NewOr = Builder.CreateOr(Z, ~(*C2)); |
| 821 | return Builder.CreateSub(RHS, NewOr, "sub"); |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 822 | } |
| 823 | return nullptr; |
| 824 | } |
| 825 | |
Sanjay Patel | 8d810fe | 2017-10-13 16:43:58 +0000 | [diff] [blame] | 826 | Instruction *InstCombiner::foldAddWithConstant(BinaryOperator &Add) { |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 827 | Value *Op0 = Add.getOperand(0), *Op1 = Add.getOperand(1); |
Sanjay Patel | 2150651 | 2017-10-13 16:29:38 +0000 | [diff] [blame] | 828 | Constant *Op1C; |
| 829 | if (!match(Op1, m_Constant(Op1C))) |
| 830 | return nullptr; |
| 831 | |
Sanjay Patel | 8fdd87f | 2018-02-28 16:36:24 +0000 | [diff] [blame] | 832 | if (Instruction *NV = foldBinOpIntoSelectOrPhi(Add)) |
Sanjay Patel | 8d810fe | 2017-10-13 16:43:58 +0000 | [diff] [blame] | 833 | return NV; |
| 834 | |
Sanjay Patel | 818b253 | 2018-07-28 16:48:44 +0000 | [diff] [blame] | 835 | Value *X, *Y; |
| 836 | |
| 837 | // add (sub X, Y), -1 --> add (not Y), X |
| 838 | if (match(Op0, m_OneUse(m_Sub(m_Value(X), m_Value(Y)))) && |
| 839 | match(Op1, m_AllOnes())) |
| 840 | return BinaryOperator::CreateAdd(Builder.CreateNot(Y), X); |
| 841 | |
Sanjay Patel | f0242de | 2017-10-13 20:29:11 +0000 | [diff] [blame] | 842 | // zext(bool) + C -> bool ? C + 1 : C |
| 843 | if (match(Op0, m_ZExt(m_Value(X))) && |
| 844 | X->getType()->getScalarSizeInBits() == 1) |
Sanjay Patel | 76ed9ea | 2017-10-13 17:00:47 +0000 | [diff] [blame] | 845 | return SelectInst::Create(X, AddOne(Op1C), Op1); |
Sanjay Patel | 2150651 | 2017-10-13 16:29:38 +0000 | [diff] [blame] | 846 | |
Sanjay Patel | f0242de | 2017-10-13 20:29:11 +0000 | [diff] [blame] | 847 | // ~X + C --> (C-1) - X |
| 848 | if (match(Op0, m_Not(m_Value(X)))) |
| 849 | return BinaryOperator::CreateSub(SubOne(Op1C), X); |
| 850 | |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 851 | const APInt *C; |
| 852 | if (!match(Op1, m_APInt(C))) |
| 853 | return nullptr; |
| 854 | |
| 855 | if (C->isSignMask()) { |
| 856 | // If wrapping is not allowed, then the addition must set the sign bit: |
| 857 | // X + (signmask) --> X | signmask |
| 858 | if (Add.hasNoSignedWrap() || Add.hasNoUnsignedWrap()) |
| 859 | return BinaryOperator::CreateOr(Op0, Op1); |
| 860 | |
| 861 | // If wrapping is allowed, then the addition flips the sign bit of LHS: |
| 862 | // X + (signmask) --> X ^ signmask |
| 863 | return BinaryOperator::CreateXor(Op0, Op1); |
| 864 | } |
| 865 | |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 866 | // Is this add the last step in a convoluted sext? |
| 867 | // add(zext(xor i16 X, -32768), -32768) --> sext X |
Sanjay Patel | 76ed9ea | 2017-10-13 17:00:47 +0000 | [diff] [blame] | 868 | Type *Ty = Add.getType(); |
Sanjay Patel | 2150651 | 2017-10-13 16:29:38 +0000 | [diff] [blame] | 869 | const APInt *C2; |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 870 | if (match(Op0, m_ZExt(m_Xor(m_Value(X), m_APInt(C2)))) && |
| 871 | C2->isMinSignedValue() && C2->sext(Ty->getScalarSizeInBits()) == *C) |
| 872 | return CastInst::Create(Instruction::SExt, X, Ty); |
| 873 | |
| 874 | // (add (zext (add nuw X, C2)), C) --> (zext (add nuw X, C2 + C)) |
Sanjay Patel | c419c9f | 2017-10-13 17:47:25 +0000 | [diff] [blame] | 875 | if (match(Op0, m_OneUse(m_ZExt(m_NUWAdd(m_Value(X), m_APInt(C2))))) && |
| 876 | C->isNegative() && C->sge(-C2->sext(C->getBitWidth()))) { |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 877 | Constant *NewC = |
| 878 | ConstantInt::get(X->getType(), *C2 + C->trunc(C2->getBitWidth())); |
| 879 | return new ZExtInst(Builder.CreateNUWAdd(X, NewC), Ty); |
| 880 | } |
| 881 | |
Sanjay Patel | 2f3ead7 | 2017-06-25 14:15:28 +0000 | [diff] [blame] | 882 | if (C->isOneValue() && Op0->hasOneUse()) { |
| 883 | // add (sext i1 X), 1 --> zext (not X) |
| 884 | // TODO: The smallest IR representation is (select X, 0, 1), and that would |
| 885 | // not require the one-use check. But we need to remove a transform in |
| 886 | // visitSelect and make sure that IR value tracking for select is equal or |
| 887 | // better than for these ops. |
| 888 | if (match(Op0, m_SExt(m_Value(X))) && |
| 889 | X->getType()->getScalarSizeInBits() == 1) |
| 890 | return new ZExtInst(Builder.CreateNot(X), Ty); |
| 891 | |
| 892 | // Shifts and add used to flip and mask off the low bit: |
| 893 | // add (ashr (shl i32 X, 31), 31), 1 --> and (not X), 1 |
| 894 | const APInt *C3; |
| 895 | if (match(Op0, m_AShr(m_Shl(m_Value(X), m_APInt(C2)), m_APInt(C3))) && |
| 896 | C2 == C3 && *C2 == Ty->getScalarSizeInBits() - 1) { |
| 897 | Value *NotX = Builder.CreateNot(X); |
| 898 | return BinaryOperator::CreateAnd(NotX, ConstantInt::get(Ty, 1)); |
| 899 | } |
Sanjay Patel | 2e069f2 | 2017-05-10 13:56:52 +0000 | [diff] [blame] | 900 | } |
| 901 | |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 902 | return nullptr; |
| 903 | } |
| 904 | |
Sanjoy Das | 6f1937b | 2018-04-26 20:52:28 +0000 | [diff] [blame] | 905 | // Matches multiplication expression Op * C where C is a constant. Returns the |
| 906 | // constant value in C and the other operand in Op. Returns true if such a |
| 907 | // match is found. |
| 908 | static bool MatchMul(Value *E, Value *&Op, APInt &C) { |
| 909 | const APInt *AI; |
| 910 | if (match(E, m_Mul(m_Value(Op), m_APInt(AI)))) { |
| 911 | C = *AI; |
| 912 | return true; |
| 913 | } |
| 914 | if (match(E, m_Shl(m_Value(Op), m_APInt(AI)))) { |
| 915 | C = APInt(AI->getBitWidth(), 1); |
| 916 | C <<= *AI; |
| 917 | return true; |
| 918 | } |
| 919 | return false; |
| 920 | } |
| 921 | |
| 922 | // Matches remainder expression Op % C where C is a constant. Returns the |
| 923 | // constant value in C and the other operand in Op. Returns the signedness of |
| 924 | // the remainder operation in IsSigned. Returns true if such a match is |
| 925 | // found. |
| 926 | static bool MatchRem(Value *E, Value *&Op, APInt &C, bool &IsSigned) { |
| 927 | const APInt *AI; |
| 928 | IsSigned = false; |
| 929 | if (match(E, m_SRem(m_Value(Op), m_APInt(AI)))) { |
| 930 | IsSigned = true; |
| 931 | C = *AI; |
| 932 | return true; |
| 933 | } |
| 934 | if (match(E, m_URem(m_Value(Op), m_APInt(AI)))) { |
| 935 | C = *AI; |
| 936 | return true; |
| 937 | } |
| 938 | if (match(E, m_And(m_Value(Op), m_APInt(AI))) && (*AI + 1).isPowerOf2()) { |
| 939 | C = *AI + 1; |
| 940 | return true; |
| 941 | } |
| 942 | return false; |
| 943 | } |
| 944 | |
| 945 | // Matches division expression Op / C with the given signedness as indicated |
| 946 | // by IsSigned, where C is a constant. Returns the constant value in C and the |
| 947 | // other operand in Op. Returns true if such a match is found. |
| 948 | static bool MatchDiv(Value *E, Value *&Op, APInt &C, bool IsSigned) { |
| 949 | const APInt *AI; |
| 950 | if (IsSigned && match(E, m_SDiv(m_Value(Op), m_APInt(AI)))) { |
| 951 | C = *AI; |
| 952 | return true; |
| 953 | } |
| 954 | if (!IsSigned) { |
| 955 | if (match(E, m_UDiv(m_Value(Op), m_APInt(AI)))) { |
| 956 | C = *AI; |
| 957 | return true; |
| 958 | } |
| 959 | if (match(E, m_LShr(m_Value(Op), m_APInt(AI)))) { |
| 960 | C = APInt(AI->getBitWidth(), 1); |
| 961 | C <<= *AI; |
| 962 | return true; |
| 963 | } |
| 964 | } |
| 965 | return false; |
| 966 | } |
| 967 | |
| 968 | // Returns whether C0 * C1 with the given signedness overflows. |
| 969 | static bool MulWillOverflow(APInt &C0, APInt &C1, bool IsSigned) { |
| 970 | bool overflow; |
| 971 | if (IsSigned) |
| 972 | (void)C0.smul_ov(C1, overflow); |
| 973 | else |
| 974 | (void)C0.umul_ov(C1, overflow); |
| 975 | return overflow; |
| 976 | } |
| 977 | |
| 978 | // Simplifies X % C0 + (( X / C0 ) % C1) * C0 to X % (C0 * C1), where (C0 * C1) |
| 979 | // does not overflow. |
| 980 | Value *InstCombiner::SimplifyAddWithRemainder(BinaryOperator &I) { |
| 981 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); |
| 982 | Value *X, *MulOpV; |
| 983 | APInt C0, MulOpC; |
| 984 | bool IsSigned; |
| 985 | // Match I = X % C0 + MulOpV * C0 |
| 986 | if (((MatchRem(LHS, X, C0, IsSigned) && MatchMul(RHS, MulOpV, MulOpC)) || |
| 987 | (MatchRem(RHS, X, C0, IsSigned) && MatchMul(LHS, MulOpV, MulOpC))) && |
| 988 | C0 == MulOpC) { |
| 989 | Value *RemOpV; |
| 990 | APInt C1; |
| 991 | bool Rem2IsSigned; |
| 992 | // Match MulOpC = RemOpV % C1 |
| 993 | if (MatchRem(MulOpV, RemOpV, C1, Rem2IsSigned) && |
| 994 | IsSigned == Rem2IsSigned) { |
| 995 | Value *DivOpV; |
| 996 | APInt DivOpC; |
| 997 | // Match RemOpV = X / C0 |
| 998 | if (MatchDiv(RemOpV, DivOpV, DivOpC, IsSigned) && X == DivOpV && |
| 999 | C0 == DivOpC && !MulWillOverflow(C0, C1, IsSigned)) { |
| 1000 | Value *NewDivisor = |
| 1001 | ConstantInt::get(X->getType()->getContext(), C0 * C1); |
| 1002 | return IsSigned ? Builder.CreateSRem(X, NewDivisor, "srem") |
| 1003 | : Builder.CreateURem(X, NewDivisor, "urem"); |
| 1004 | } |
| 1005 | } |
| 1006 | } |
| 1007 | |
| 1008 | return nullptr; |
| 1009 | } |
| 1010 | |
Roman Lebedev | cbf8446 | 2018-06-06 19:38:27 +0000 | [diff] [blame] | 1011 | /// Fold |
| 1012 | /// (1 << NBits) - 1 |
| 1013 | /// Into: |
| 1014 | /// ~(-(1 << NBits)) |
| 1015 | /// Because a 'not' is better for bit-tracking analysis and other transforms |
| 1016 | /// than an 'add'. The new shl is always nsw, and is nuw if old `and` was. |
| 1017 | static Instruction *canonicalizeLowbitMask(BinaryOperator &I, |
| 1018 | InstCombiner::BuilderTy &Builder) { |
| 1019 | Value *NBits; |
| 1020 | if (!match(&I, m_Add(m_OneUse(m_Shl(m_One(), m_Value(NBits))), m_AllOnes()))) |
| 1021 | return nullptr; |
| 1022 | |
| 1023 | Constant *MinusOne = Constant::getAllOnesValue(NBits->getType()); |
| 1024 | Value *NotMask = Builder.CreateShl(MinusOne, NBits, "notmask"); |
| 1025 | // Be wary of constant folding. |
| 1026 | if (auto *BOp = dyn_cast<BinaryOperator>(NotMask)) { |
| 1027 | // Always NSW. But NUW propagates from `add`. |
| 1028 | BOp->setHasNoSignedWrap(); |
| 1029 | BOp->setHasNoUnsignedWrap(I.hasNoUnsignedWrap()); |
| 1030 | } |
| 1031 | |
| 1032 | return BinaryOperator::CreateNot(NotMask, I.getName()); |
| 1033 | } |
| 1034 | |
Dinesh Dwivedi | adc0773 | 2014-06-27 07:47:35 +0000 | [diff] [blame] | 1035 | Instruction *InstCombiner::visitAdd(BinaryOperator &I) { |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1036 | if (Value *V = SimplifyAddInst(I.getOperand(0), I.getOperand(1), |
| 1037 | I.hasNoSignedWrap(), I.hasNoUnsignedWrap(), |
| 1038 | SQ.getWithInstruction(&I))) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1039 | return replaceInstUsesWith(I, V); |
Serge Pavlov | 9ef66a8 | 2014-05-11 08:46:12 +0000 | [diff] [blame] | 1040 | |
Sanjay Patel | 70043b7 | 2018-07-13 01:18:07 +0000 | [diff] [blame] | 1041 | if (SimplifyAssociativeOrCommutative(I)) |
| 1042 | return &I; |
| 1043 | |
Sanjay Patel | 79dceb2 | 2018-10-03 15:20:58 +0000 | [diff] [blame^] | 1044 | if (Instruction *X = foldVectorBinop(I)) |
Sanjay Patel | bbc6d60 | 2018-06-02 16:27:44 +0000 | [diff] [blame] | 1045 | return X; |
| 1046 | |
Eugene Zelenko | 7f0f9bc | 2017-10-24 21:24:53 +0000 | [diff] [blame] | 1047 | // (A*B)+(A*C) -> A*(B+C) etc |
Duncan Sands | fbb9ac3 | 2010-12-22 13:36:08 +0000 | [diff] [blame] | 1048 | if (Value *V = SimplifyUsingDistributiveLaws(I)) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1049 | return replaceInstUsesWith(I, V); |
Duncan Sands | fbb9ac3 | 2010-12-22 13:36:08 +0000 | [diff] [blame] | 1050 | |
Sanjay Patel | 8d810fe | 2017-10-13 16:43:58 +0000 | [diff] [blame] | 1051 | if (Instruction *X = foldAddWithConstant(I)) |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 1052 | return X; |
Sanjay Patel | 53c5c3d | 2017-02-18 22:20:09 +0000 | [diff] [blame] | 1053 | |
Sanjay Patel | 4133d4a | 2017-05-10 00:07:16 +0000 | [diff] [blame] | 1054 | // FIXME: This should be moved into the above helper function to allow these |
Sanjay Patel | 2150651 | 2017-10-13 16:29:38 +0000 | [diff] [blame] | 1055 | // transforms for general constant or constant splat vectors. |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1056 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); |
Sanjay Patel | 2118952 | 2017-10-13 18:32:53 +0000 | [diff] [blame] | 1057 | Type *Ty = I.getType(); |
Sanjay Patel | 79acd2a | 2016-07-16 18:29:26 +0000 | [diff] [blame] | 1058 | if (ConstantInt *CI = dyn_cast<ConstantInt>(RHS)) { |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1059 | Value *XorLHS = nullptr; ConstantInt *XorRHS = nullptr; |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1060 | if (match(LHS, m_Xor(m_Value(XorLHS), m_ConstantInt(XorRHS)))) { |
Sanjay Patel | 2118952 | 2017-10-13 18:32:53 +0000 | [diff] [blame] | 1061 | unsigned TySizeBits = Ty->getScalarSizeInBits(); |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1062 | const APInt &RHSVal = CI->getValue(); |
Eli Friedman | a2cc287 | 2010-01-31 04:29:12 +0000 | [diff] [blame] | 1063 | unsigned ExtendAmt = 0; |
| 1064 | // If we have ADD(XOR(AND(X, 0xFF), 0x80), 0xF..F80), it's a sext. |
| 1065 | // If we have ADD(XOR(AND(X, 0xFF), 0xF..F80), 0x80), it's a sext. |
| 1066 | if (XorRHS->getValue() == -RHSVal) { |
| 1067 | if (RHSVal.isPowerOf2()) |
| 1068 | ExtendAmt = TySizeBits - RHSVal.logBase2() - 1; |
| 1069 | else if (XorRHS->getValue().isPowerOf2()) |
| 1070 | ExtendAmt = TySizeBits - XorRHS->getValue().logBase2() - 1; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1071 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1072 | |
Eli Friedman | a2cc287 | 2010-01-31 04:29:12 +0000 | [diff] [blame] | 1073 | if (ExtendAmt) { |
| 1074 | APInt Mask = APInt::getHighBitsSet(TySizeBits, ExtendAmt); |
Hal Finkel | 60db058 | 2014-09-07 18:57:58 +0000 | [diff] [blame] | 1075 | if (!MaskedValueIsZero(XorLHS, Mask, 0, &I)) |
Eli Friedman | a2cc287 | 2010-01-31 04:29:12 +0000 | [diff] [blame] | 1076 | ExtendAmt = 0; |
| 1077 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1078 | |
Eli Friedman | a2cc287 | 2010-01-31 04:29:12 +0000 | [diff] [blame] | 1079 | if (ExtendAmt) { |
Sanjay Patel | 2118952 | 2017-10-13 18:32:53 +0000 | [diff] [blame] | 1080 | Constant *ShAmt = ConstantInt::get(Ty, ExtendAmt); |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1081 | Value *NewShl = Builder.CreateShl(XorLHS, ShAmt, "sext"); |
Eli Friedman | a2cc287 | 2010-01-31 04:29:12 +0000 | [diff] [blame] | 1082 | return BinaryOperator::CreateAShr(NewShl, ShAmt); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1083 | } |
Benjamin Kramer | b16bd77 | 2011-12-24 17:31:53 +0000 | [diff] [blame] | 1084 | |
| 1085 | // If this is a xor that was canonicalized from a sub, turn it back into |
| 1086 | // a sub and fuse this add with it. |
| 1087 | if (LHS->hasOneUse() && (XorRHS->getValue()+1).isPowerOf2()) { |
Craig Topper | 8205a1a | 2017-05-24 16:53:07 +0000 | [diff] [blame] | 1088 | KnownBits LHSKnown = computeKnownBits(XorLHS, 0, &I); |
Craig Topper | b45eabc | 2017-04-26 16:39:58 +0000 | [diff] [blame] | 1089 | if ((XorRHS->getValue() | LHSKnown.Zero).isAllOnesValue()) |
Benjamin Kramer | b16bd77 | 2011-12-24 17:31:53 +0000 | [diff] [blame] | 1090 | return BinaryOperator::CreateSub(ConstantExpr::getAdd(XorRHS, CI), |
| 1091 | XorLHS); |
| 1092 | } |
Craig Topper | bcfd2d1 | 2017-04-20 16:56:25 +0000 | [diff] [blame] | 1093 | // (X + signmask) + C could have gotten canonicalized to (X^signmask) + C, |
| 1094 | // transform them into (X + (signmask ^ C)) |
| 1095 | if (XorRHS->getValue().isSignMask()) |
Craig Topper | eafbd57 | 2015-12-21 01:02:28 +0000 | [diff] [blame] | 1096 | return BinaryOperator::CreateAdd(XorLHS, |
| 1097 | ConstantExpr::getXor(XorRHS, CI)); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1098 | } |
| 1099 | } |
| 1100 | |
Sanjay Patel | 2118952 | 2017-10-13 18:32:53 +0000 | [diff] [blame] | 1101 | if (Ty->isIntOrIntVectorTy(1)) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1102 | return BinaryOperator::CreateXor(LHS, RHS); |
| 1103 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1104 | // X + X --> X << 1 |
Chris Lattner | d406764 | 2011-02-17 20:55:29 +0000 | [diff] [blame] | 1105 | if (LHS == RHS) { |
Sanjay Patel | 2118952 | 2017-10-13 18:32:53 +0000 | [diff] [blame] | 1106 | auto *Shl = BinaryOperator::CreateShl(LHS, ConstantInt::get(Ty, 1)); |
| 1107 | Shl->setHasNoSignedWrap(I.hasNoSignedWrap()); |
| 1108 | Shl->setHasNoUnsignedWrap(I.hasNoUnsignedWrap()); |
| 1109 | return Shl; |
Chris Lattner | 5592071 | 2011-02-17 02:23:02 +0000 | [diff] [blame] | 1110 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1111 | |
Sanjay Patel | b869f76 | 2017-10-13 21:28:50 +0000 | [diff] [blame] | 1112 | Value *A, *B; |
| 1113 | if (match(LHS, m_Neg(m_Value(A)))) { |
| 1114 | // -A + -B --> -(A + B) |
| 1115 | if (match(RHS, m_Neg(m_Value(B)))) |
| 1116 | return BinaryOperator::CreateNeg(Builder.CreateAdd(A, B)); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1117 | |
Sanjay Patel | b869f76 | 2017-10-13 21:28:50 +0000 | [diff] [blame] | 1118 | // -A + B --> B - A |
| 1119 | return BinaryOperator::CreateSub(RHS, A); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1120 | } |
| 1121 | |
| 1122 | // A + -B --> A - B |
Sanjay Patel | b869f76 | 2017-10-13 21:28:50 +0000 | [diff] [blame] | 1123 | if (match(RHS, m_Neg(m_Value(B)))) |
| 1124 | return BinaryOperator::CreateSub(LHS, B); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1125 | |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 1126 | if (Value *V = checkForNegativeOperand(I, Builder)) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1127 | return replaceInstUsesWith(I, V); |
Dinesh Dwivedi | 562fd75 | 2014-06-19 10:36:52 +0000 | [diff] [blame] | 1128 | |
Gil Rapaport | da2e2ca | 2018-06-26 05:31:18 +0000 | [diff] [blame] | 1129 | // (A + 1) + ~B --> A - B |
| 1130 | // ~B + (A + 1) --> A - B |
| 1131 | if (match(&I, m_c_BinOp(m_Add(m_Value(A), m_One()), m_Not(m_Value(B))))) |
| 1132 | return BinaryOperator::CreateSub(A, B); |
| 1133 | |
Sanjoy Das | 6f1937b | 2018-04-26 20:52:28 +0000 | [diff] [blame] | 1134 | // X % C0 + (( X / C0 ) % C1) * C0 => X % (C0 * C1) |
| 1135 | if (Value *V = SimplifyAddWithRemainder(I)) return replaceInstUsesWith(I, V); |
| 1136 | |
Sylvestre Ledru | 91ce36c | 2012-09-27 10:14:43 +0000 | [diff] [blame] | 1137 | // A+B --> A|B iff A and B have no bits set in common. |
Daniel Jasper | aec2fa3 | 2016-12-19 08:22:17 +0000 | [diff] [blame] | 1138 | if (haveNoCommonBitsSet(LHS, RHS, DL, &AC, &I, &DT)) |
Jingyue Wu | ca32190 | 2015-05-14 23:53:19 +0000 | [diff] [blame] | 1139 | return BinaryOperator::CreateOr(LHS, RHS); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1140 | |
Sanjay Patel | 79acd2a | 2016-07-16 18:29:26 +0000 | [diff] [blame] | 1141 | // FIXME: We already did a check for ConstantInt RHS above this. |
| 1142 | // FIXME: Is this pattern covered by another fold? No regression tests fail on |
| 1143 | // removal. |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1144 | if (ConstantInt *CRHS = dyn_cast<ConstantInt>(RHS)) { |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1145 | // (X & FF00) + xx00 -> (X+xx00) & FF00 |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1146 | Value *X; |
| 1147 | ConstantInt *C2; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1148 | if (LHS->hasOneUse() && |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1149 | match(LHS, m_And(m_Value(X), m_ConstantInt(C2))) && |
| 1150 | CRHS->getValue() == (CRHS->getValue() & C2->getValue())) { |
| 1151 | // See if all bits from the first bit set in the Add RHS up are included |
| 1152 | // in the mask. First, get the rightmost bit. |
| 1153 | const APInt &AddRHSV = CRHS->getValue(); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1154 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1155 | // Form a mask of all bits from the lowest bit added through the top. |
| 1156 | APInt AddRHSHighBits(~((AddRHSV & -AddRHSV)-1)); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1157 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1158 | // See if the and mask includes all of these bits. |
| 1159 | APInt AddRHSHighBitsAnd(AddRHSHighBits & C2->getValue()); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1160 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1161 | if (AddRHSHighBits == AddRHSHighBitsAnd) { |
| 1162 | // Okay, the xform is safe. Insert the new add pronto. |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1163 | Value *NewAdd = Builder.CreateAdd(X, CRHS, LHS->getName()); |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1164 | return BinaryOperator::CreateAnd(NewAdd, C2); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1165 | } |
| 1166 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1167 | } |
| 1168 | |
| 1169 | // add (select X 0 (sub n A)) A --> select X A n |
| 1170 | { |
| 1171 | SelectInst *SI = dyn_cast<SelectInst>(LHS); |
| 1172 | Value *A = RHS; |
| 1173 | if (!SI) { |
| 1174 | SI = dyn_cast<SelectInst>(RHS); |
| 1175 | A = LHS; |
| 1176 | } |
| 1177 | if (SI && SI->hasOneUse()) { |
| 1178 | Value *TV = SI->getTrueValue(); |
| 1179 | Value *FV = SI->getFalseValue(); |
| 1180 | Value *N; |
| 1181 | |
| 1182 | // Can we fold the add into the argument of the select? |
| 1183 | // We check both true and false select arguments for a matching subtract. |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1184 | if (match(FV, m_Zero()) && match(TV, m_Sub(m_Value(N), m_Specific(A)))) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1185 | // Fold the add into the true select value. |
| 1186 | return SelectInst::Create(SI->getCondition(), N, A); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1187 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1188 | if (match(TV, m_Zero()) && match(FV, m_Sub(m_Value(N), m_Specific(A)))) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1189 | // Fold the add into the false select value. |
| 1190 | return SelectInst::Create(SI->getCondition(), A, N); |
| 1191 | } |
| 1192 | } |
| 1193 | |
Sanjay Patel | 90a3634 | 2018-09-14 22:23:35 +0000 | [diff] [blame] | 1194 | if (Instruction *Ext = narrowMathIfNoOverflow(I)) |
Sanjay Patel | 2426eb4 | 2018-09-14 20:40:46 +0000 | [diff] [blame] | 1195 | return Ext; |
David Majnemer | a1cfd7c | 2016-12-30 00:28:58 +0000 | [diff] [blame] | 1196 | |
David Majnemer | ab07f00 | 2014-08-11 22:32:02 +0000 | [diff] [blame] | 1197 | // (add (xor A, B) (and A, B)) --> (or A, B) |
Sanjay Patel | 28b3aa3 | 2017-10-13 20:12:21 +0000 | [diff] [blame] | 1198 | // (add (and A, B) (xor A, B)) --> (or A, B) |
Roman Lebedev | 6959b8e | 2018-04-27 21:23:20 +0000 | [diff] [blame] | 1199 | if (match(&I, m_c_BinOp(m_Xor(m_Value(A), m_Value(B)), |
| 1200 | m_c_And(m_Deferred(A), m_Deferred(B))))) |
Sanjay Patel | 28b3aa3 | 2017-10-13 20:12:21 +0000 | [diff] [blame] | 1201 | return BinaryOperator::CreateOr(A, B); |
Chad Rosier | 7813dce | 2012-04-26 23:29:14 +0000 | [diff] [blame] | 1202 | |
David Majnemer | ab07f00 | 2014-08-11 22:32:02 +0000 | [diff] [blame] | 1203 | // (add (or A, B) (and A, B)) --> (add A, B) |
Sanjay Patel | 28b3aa3 | 2017-10-13 20:12:21 +0000 | [diff] [blame] | 1204 | // (add (and A, B) (or A, B)) --> (add A, B) |
Roman Lebedev | 6959b8e | 2018-04-27 21:23:20 +0000 | [diff] [blame] | 1205 | if (match(&I, m_c_BinOp(m_Or(m_Value(A), m_Value(B)), |
| 1206 | m_c_And(m_Deferred(A), m_Deferred(B))))) { |
Sanjay Patel | 28b3aa3 | 2017-10-13 20:12:21 +0000 | [diff] [blame] | 1207 | I.setOperand(0, A); |
| 1208 | I.setOperand(1, B); |
| 1209 | return &I; |
David Majnemer | ab07f00 | 2014-08-11 22:32:02 +0000 | [diff] [blame] | 1210 | } |
| 1211 | |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1212 | // TODO(jingyue): Consider willNotOverflowSignedAdd and |
Craig Topper | bb97372 | 2017-05-15 02:44:08 +0000 | [diff] [blame] | 1213 | // willNotOverflowUnsignedAdd to reduce the number of invocations of |
Jingyue Wu | 33bd53d | 2014-06-17 00:42:07 +0000 | [diff] [blame] | 1214 | // computeKnownBits. |
Sanjay Patel | 70043b7 | 2018-07-13 01:18:07 +0000 | [diff] [blame] | 1215 | bool Changed = false; |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1216 | if (!I.hasNoSignedWrap() && willNotOverflowSignedAdd(LHS, RHS, I)) { |
Rafael Espindola | d1a2c2d | 2014-06-02 22:01:04 +0000 | [diff] [blame] | 1217 | Changed = true; |
| 1218 | I.setHasNoSignedWrap(true); |
| 1219 | } |
Craig Topper | bb97372 | 2017-05-15 02:44:08 +0000 | [diff] [blame] | 1220 | if (!I.hasNoUnsignedWrap() && willNotOverflowUnsignedAdd(LHS, RHS, I)) { |
Jingyue Wu | 33bd53d | 2014-06-17 00:42:07 +0000 | [diff] [blame] | 1221 | Changed = true; |
| 1222 | I.setHasNoUnsignedWrap(true); |
| 1223 | } |
Rafael Espindola | d1a2c2d | 2014-06-02 22:01:04 +0000 | [diff] [blame] | 1224 | |
Roman Lebedev | cbf8446 | 2018-06-06 19:38:27 +0000 | [diff] [blame] | 1225 | if (Instruction *V = canonicalizeLowbitMask(I, Builder)) |
| 1226 | return V; |
| 1227 | |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1228 | return Changed ? &I : nullptr; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1229 | } |
| 1230 | |
Sanjay Patel | dc185ee | 2018-08-12 15:48:26 +0000 | [diff] [blame] | 1231 | /// Factor a common operand out of fadd/fsub of fmul/fdiv. |
| 1232 | static Instruction *factorizeFAddFSub(BinaryOperator &I, |
| 1233 | InstCombiner::BuilderTy &Builder) { |
| 1234 | assert((I.getOpcode() == Instruction::FAdd || |
| 1235 | I.getOpcode() == Instruction::FSub) && "Expecting fadd/fsub"); |
| 1236 | assert(I.hasAllowReassoc() && I.hasNoSignedZeros() && |
| 1237 | "FP factorization requires FMF"); |
| 1238 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); |
| 1239 | Value *X, *Y, *Z; |
| 1240 | bool IsFMul; |
| 1241 | if ((match(Op0, m_OneUse(m_FMul(m_Value(X), m_Value(Z)))) && |
| 1242 | match(Op1, m_OneUse(m_c_FMul(m_Value(Y), m_Specific(Z))))) || |
| 1243 | (match(Op0, m_OneUse(m_FMul(m_Value(Z), m_Value(X)))) && |
| 1244 | match(Op1, m_OneUse(m_c_FMul(m_Value(Y), m_Specific(Z)))))) |
| 1245 | IsFMul = true; |
| 1246 | else if (match(Op0, m_OneUse(m_FDiv(m_Value(X), m_Value(Z)))) && |
| 1247 | match(Op1, m_OneUse(m_FDiv(m_Value(Y), m_Specific(Z))))) |
| 1248 | IsFMul = false; |
| 1249 | else |
| 1250 | return nullptr; |
| 1251 | |
| 1252 | // (X * Z) + (Y * Z) --> (X + Y) * Z |
| 1253 | // (X * Z) - (Y * Z) --> (X - Y) * Z |
| 1254 | // (X / Z) + (Y / Z) --> (X + Y) / Z |
| 1255 | // (X / Z) - (Y / Z) --> (X - Y) / Z |
| 1256 | bool IsFAdd = I.getOpcode() == Instruction::FAdd; |
| 1257 | Value *XY = IsFAdd ? Builder.CreateFAddFMF(X, Y, &I) |
| 1258 | : Builder.CreateFSubFMF(X, Y, &I); |
| 1259 | |
| 1260 | // Bail out if we just created a denormal constant. |
| 1261 | // TODO: This is copied from a previous implementation. Is it necessary? |
| 1262 | const APFloat *C; |
| 1263 | if (match(XY, m_APFloat(C)) && !C->isNormal()) |
| 1264 | return nullptr; |
| 1265 | |
| 1266 | return IsFMul ? BinaryOperator::CreateFMulFMF(XY, Z, &I) |
| 1267 | : BinaryOperator::CreateFDivFMF(XY, Z, &I); |
| 1268 | } |
| 1269 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1270 | Instruction *InstCombiner::visitFAdd(BinaryOperator &I) { |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1271 | if (Value *V = SimplifyFAddInst(I.getOperand(0), I.getOperand(1), |
| 1272 | I.getFastMathFlags(), |
Craig Topper | a420562 | 2017-06-09 03:21:29 +0000 | [diff] [blame] | 1273 | SQ.getWithInstruction(&I))) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1274 | return replaceInstUsesWith(I, V); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1275 | |
Sanjay Patel | 70043b7 | 2018-07-13 01:18:07 +0000 | [diff] [blame] | 1276 | if (SimplifyAssociativeOrCommutative(I)) |
| 1277 | return &I; |
| 1278 | |
Sanjay Patel | 79dceb2 | 2018-10-03 15:20:58 +0000 | [diff] [blame^] | 1279 | if (Instruction *X = foldVectorBinop(I)) |
Sanjay Patel | bbc6d60 | 2018-06-02 16:27:44 +0000 | [diff] [blame] | 1280 | return X; |
| 1281 | |
Sanjay Patel | 8fdd87f | 2018-02-28 16:36:24 +0000 | [diff] [blame] | 1282 | if (Instruction *FoldedFAdd = foldBinOpIntoSelectOrPhi(I)) |
| 1283 | return FoldedFAdd; |
Michael Ilseman | e2754dc | 2012-12-14 22:08:26 +0000 | [diff] [blame] | 1284 | |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1285 | Value *LHS = I.getOperand(0), *RHS = I.getOperand(1); |
Sanjay Patel | 1170daa | 2018-04-16 14:13:57 +0000 | [diff] [blame] | 1286 | Value *X; |
| 1287 | // (-X) + Y --> Y - X |
| 1288 | if (match(LHS, m_FNeg(m_Value(X)))) |
| 1289 | return BinaryOperator::CreateFSubFMF(RHS, X, &I); |
| 1290 | // Y + (-X) --> Y - X |
| 1291 | if (match(RHS, m_FNeg(m_Value(X)))) |
| 1292 | return BinaryOperator::CreateFSubFMF(LHS, X, &I); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1293 | |
Dan Gohman | 6f34abd | 2010-03-02 01:11:08 +0000 | [diff] [blame] | 1294 | // Check for (fadd double (sitofp x), y), see if we can merge this into an |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1295 | // integer add followed by a promotion. |
| 1296 | if (SIToFPInst *LHSConv = dyn_cast<SIToFPInst>(LHS)) { |
Artur Pilipenko | 4cc6130 | 2017-03-21 11:32:15 +0000 | [diff] [blame] | 1297 | Value *LHSIntVal = LHSConv->getOperand(0); |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1298 | Type *FPType = LHSConv->getType(); |
| 1299 | |
| 1300 | // TODO: This check is overly conservative. In many cases known bits |
| 1301 | // analysis can tell us that the result of the addition has less significant |
| 1302 | // bits than the integer type can hold. |
| 1303 | auto IsValidPromotion = [](Type *FTy, Type *ITy) { |
Artur Pilipenko | 0632bdc | 2017-04-22 07:24:52 +0000 | [diff] [blame] | 1304 | Type *FScalarTy = FTy->getScalarType(); |
| 1305 | Type *IScalarTy = ITy->getScalarType(); |
| 1306 | |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1307 | // Do we have enough bits in the significand to represent the result of |
| 1308 | // the integer addition? |
| 1309 | unsigned MaxRepresentableBits = |
Artur Pilipenko | 0632bdc | 2017-04-22 07:24:52 +0000 | [diff] [blame] | 1310 | APFloat::semanticsPrecision(FScalarTy->getFltSemantics()); |
| 1311 | return IScalarTy->getIntegerBitWidth() <= MaxRepresentableBits; |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1312 | }; |
Artur Pilipenko | 4cc6130 | 2017-03-21 11:32:15 +0000 | [diff] [blame] | 1313 | |
Dan Gohman | 6f34abd | 2010-03-02 01:11:08 +0000 | [diff] [blame] | 1314 | // (fadd double (sitofp x), fpcst) --> (sitofp (add int x, intcst)) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1315 | // ... if the constant fits in the integer value. This is useful for things |
| 1316 | // like (double)(x & 1234) + 4.0 -> (double)((X & 1234)+4) which no longer |
| 1317 | // requires a constant pool load, and generally allows the add to be better |
| 1318 | // instcombined. |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1319 | if (ConstantFP *CFP = dyn_cast<ConstantFP>(RHS)) |
| 1320 | if (IsValidPromotion(FPType, LHSIntVal->getType())) { |
| 1321 | Constant *CI = |
| 1322 | ConstantExpr::getFPToSI(CFP, LHSIntVal->getType()); |
| 1323 | if (LHSConv->hasOneUse() && |
| 1324 | ConstantExpr::getSIToFP(CI, I.getType()) == CFP && |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1325 | willNotOverflowSignedAdd(LHSIntVal, CI, I)) { |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1326 | // Insert the new integer add. |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1327 | Value *NewAdd = Builder.CreateNSWAdd(LHSIntVal, CI, "addconv"); |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1328 | return new SIToFPInst(NewAdd, I.getType()); |
| 1329 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1330 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1331 | |
Dan Gohman | 6f34abd | 2010-03-02 01:11:08 +0000 | [diff] [blame] | 1332 | // (fadd double (sitofp x), (sitofp y)) --> (sitofp (add int x, y)) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1333 | if (SIToFPInst *RHSConv = dyn_cast<SIToFPInst>(RHS)) { |
Artur Pilipenko | 4cc6130 | 2017-03-21 11:32:15 +0000 | [diff] [blame] | 1334 | Value *RHSIntVal = RHSConv->getOperand(0); |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1335 | // It's enough to check LHS types only because we require int types to |
| 1336 | // be the same for this transform. |
| 1337 | if (IsValidPromotion(FPType, LHSIntVal->getType())) { |
| 1338 | // Only do this if x/y have the same type, if at least one of them has a |
| 1339 | // single use (so we don't increase the number of int->fp conversions), |
| 1340 | // and if the integer add will not overflow. |
| 1341 | if (LHSIntVal->getType() == RHSIntVal->getType() && |
| 1342 | (LHSConv->hasOneUse() || RHSConv->hasOneUse()) && |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1343 | willNotOverflowSignedAdd(LHSIntVal, RHSIntVal, I)) { |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1344 | // Insert the new integer add. |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1345 | Value *NewAdd = Builder.CreateNSWAdd(LHSIntVal, RHSIntVal, "addconv"); |
Artur Pilipenko | 134d94f | 2017-04-21 18:45:25 +0000 | [diff] [blame] | 1346 | return new SIToFPInst(NewAdd, I.getType()); |
| 1347 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1348 | } |
| 1349 | } |
| 1350 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1351 | |
Quentin Colombet | aa103b3 | 2017-09-20 17:32:16 +0000 | [diff] [blame] | 1352 | // Handle specials cases for FAdd with selects feeding the operation |
| 1353 | if (Value *V = SimplifySelectsFeedingBinaryOp(I, LHS, RHS)) |
| 1354 | return replaceInstUsesWith(I, V); |
Jean-Luc Duprat | 3e4fc3e | 2013-05-06 16:55:50 +0000 | [diff] [blame] | 1355 | |
Warren Ristow | 8b2f27c | 2018-04-14 19:18:28 +0000 | [diff] [blame] | 1356 | if (I.hasAllowReassoc() && I.hasNoSignedZeros()) { |
Sanjay Patel | dc185ee | 2018-08-12 15:48:26 +0000 | [diff] [blame] | 1357 | if (Instruction *F = factorizeFAddFSub(I, Builder)) |
| 1358 | return F; |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 1359 | if (Value *V = FAddCombine(Builder).simplify(&I)) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1360 | return replaceInstUsesWith(I, V); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 1361 | } |
| 1362 | |
Sanjay Patel | 70043b7 | 2018-07-13 01:18:07 +0000 | [diff] [blame] | 1363 | return nullptr; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1364 | } |
| 1365 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1366 | /// Optimize pointer differences into the same array into a size. Consider: |
| 1367 | /// &A[10] - &A[0]: we should compile this to "10". LHS/RHS are the pointer |
| 1368 | /// operands to the ptrtoint instructions for the LHS/RHS of the subtract. |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1369 | Value *InstCombiner::OptimizePointerDifference(Value *LHS, Value *RHS, |
Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 1370 | Type *Ty) { |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1371 | // If LHS is a gep based on RHS or RHS is a gep based on LHS, we can optimize |
| 1372 | // this. |
| 1373 | bool Swapped = false; |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1374 | GEPOperator *GEP1 = nullptr, *GEP2 = nullptr; |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1375 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1376 | // For now we require one side to be the base pointer "A" or a constant |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1377 | // GEP derived from it. |
| 1378 | if (GEPOperator *LHSGEP = dyn_cast<GEPOperator>(LHS)) { |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1379 | // (gep X, ...) - X |
| 1380 | if (LHSGEP->getOperand(0) == RHS) { |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1381 | GEP1 = LHSGEP; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1382 | Swapped = false; |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1383 | } else if (GEPOperator *RHSGEP = dyn_cast<GEPOperator>(RHS)) { |
| 1384 | // (gep X, ...) - (gep X, ...) |
| 1385 | if (LHSGEP->getOperand(0)->stripPointerCasts() == |
| 1386 | RHSGEP->getOperand(0)->stripPointerCasts()) { |
| 1387 | GEP2 = RHSGEP; |
| 1388 | GEP1 = LHSGEP; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1389 | Swapped = false; |
| 1390 | } |
| 1391 | } |
| 1392 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1393 | |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1394 | if (GEPOperator *RHSGEP = dyn_cast<GEPOperator>(RHS)) { |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1395 | // X - (gep X, ...) |
| 1396 | if (RHSGEP->getOperand(0) == LHS) { |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1397 | GEP1 = RHSGEP; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1398 | Swapped = true; |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1399 | } else if (GEPOperator *LHSGEP = dyn_cast<GEPOperator>(LHS)) { |
| 1400 | // (gep X, ...) - (gep X, ...) |
| 1401 | if (RHSGEP->getOperand(0)->stripPointerCasts() == |
| 1402 | LHSGEP->getOperand(0)->stripPointerCasts()) { |
| 1403 | GEP2 = LHSGEP; |
| 1404 | GEP1 = RHSGEP; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1405 | Swapped = true; |
| 1406 | } |
| 1407 | } |
| 1408 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1409 | |
Hiroshi Yamauchi | 6085521 | 2017-07-27 18:27:11 +0000 | [diff] [blame] | 1410 | if (!GEP1) |
| 1411 | // No GEP found. |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1412 | return nullptr; |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1413 | |
Hiroshi Yamauchi | 6085521 | 2017-07-27 18:27:11 +0000 | [diff] [blame] | 1414 | if (GEP2) { |
| 1415 | // (gep X, ...) - (gep X, ...) |
| 1416 | // |
| 1417 | // Avoid duplicating the arithmetic if there are more than one non-constant |
| 1418 | // indices between the two GEPs and either GEP has a non-constant index and |
| 1419 | // multiple users. If zero non-constant index, the result is a constant and |
| 1420 | // there is no duplication. If one non-constant index, the result is an add |
| 1421 | // or sub with a constant, which is no larger than the original code, and |
| 1422 | // there's no duplicated arithmetic, even if either GEP has multiple |
| 1423 | // users. If more than one non-constant indices combined, as long as the GEP |
| 1424 | // with at least one non-constant index doesn't have multiple users, there |
| 1425 | // is no duplication. |
| 1426 | unsigned NumNonConstantIndices1 = GEP1->countNonConstantIndices(); |
| 1427 | unsigned NumNonConstantIndices2 = GEP2->countNonConstantIndices(); |
| 1428 | if (NumNonConstantIndices1 + NumNonConstantIndices2 > 1 && |
| 1429 | ((NumNonConstantIndices1 > 0 && !GEP1->hasOneUse()) || |
| 1430 | (NumNonConstantIndices2 > 0 && !GEP2->hasOneUse()))) { |
| 1431 | return nullptr; |
| 1432 | } |
| 1433 | } |
| 1434 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1435 | // Emit the offset of the GEP and an intptr_t. |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1436 | Value *Result = EmitGEPOffset(GEP1); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1437 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1438 | // If we had a constant expression GEP on the other side offsetting the |
| 1439 | // pointer, subtract it from the offset we have. |
Benjamin Kramer | 7746eb6 | 2012-02-20 14:34:57 +0000 | [diff] [blame] | 1440 | if (GEP2) { |
| 1441 | Value *Offset = EmitGEPOffset(GEP2); |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1442 | Result = Builder.CreateSub(Result, Offset); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1443 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1444 | |
| 1445 | // If we have p - gep(p, ...) then we have to negate the result. |
| 1446 | if (Swapped) |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1447 | Result = Builder.CreateNeg(Result, "diff.neg"); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1448 | |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1449 | return Builder.CreateIntCast(Result, Ty, true); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1450 | } |
| 1451 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1452 | Instruction *InstCombiner::visitSub(BinaryOperator &I) { |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1453 | if (Value *V = SimplifySubInst(I.getOperand(0), I.getOperand(1), |
| 1454 | I.hasNoSignedWrap(), I.hasNoUnsignedWrap(), |
| 1455 | SQ.getWithInstruction(&I))) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1456 | return replaceInstUsesWith(I, V); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1457 | |
Sanjay Patel | 79dceb2 | 2018-10-03 15:20:58 +0000 | [diff] [blame^] | 1458 | if (Instruction *X = foldVectorBinop(I)) |
Sanjay Patel | bbc6d60 | 2018-06-02 16:27:44 +0000 | [diff] [blame] | 1459 | return X; |
| 1460 | |
Duncan Sands | fbb9ac3 | 2010-12-22 13:36:08 +0000 | [diff] [blame] | 1461 | // (A*B)-(A*C) -> A*(B-C) etc |
| 1462 | if (Value *V = SimplifyUsingDistributiveLaws(I)) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1463 | return replaceInstUsesWith(I, V); |
Duncan Sands | fbb9ac3 | 2010-12-22 13:36:08 +0000 | [diff] [blame] | 1464 | |
David Majnemer | a92687d | 2014-07-31 04:49:29 +0000 | [diff] [blame] | 1465 | // If this is a 'B = x-(-A)', change to B = x+A. |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1466 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1467 | if (Value *V = dyn_castNegVal(Op1)) { |
| 1468 | BinaryOperator *Res = BinaryOperator::CreateAdd(Op0, V); |
David Majnemer | a92687d | 2014-07-31 04:49:29 +0000 | [diff] [blame] | 1469 | |
| 1470 | if (const auto *BO = dyn_cast<BinaryOperator>(Op1)) { |
| 1471 | assert(BO->getOpcode() == Instruction::Sub && |
| 1472 | "Expected a subtraction operator!"); |
| 1473 | if (BO->hasNoSignedWrap() && I.hasNoSignedWrap()) |
| 1474 | Res->setHasNoSignedWrap(true); |
David Majnemer | 0e6c986 | 2014-08-22 16:41:23 +0000 | [diff] [blame] | 1475 | } else { |
| 1476 | if (cast<Constant>(Op1)->isNotMinSignedValue() && I.hasNoSignedWrap()) |
| 1477 | Res->setHasNoSignedWrap(true); |
David Majnemer | a92687d | 2014-07-31 04:49:29 +0000 | [diff] [blame] | 1478 | } |
| 1479 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1480 | return Res; |
| 1481 | } |
| 1482 | |
Craig Topper | fde4723 | 2017-07-09 07:04:03 +0000 | [diff] [blame] | 1483 | if (I.getType()->isIntOrIntVectorTy(1)) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1484 | return BinaryOperator::CreateXor(Op0, Op1); |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1485 | |
| 1486 | // Replace (-1 - A) with (~A). |
| 1487 | if (match(Op0, m_AllOnes())) |
| 1488 | return BinaryOperator::CreateNot(Op1); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1489 | |
Sanjay Patel | 1a8d5c3 | 2018-03-03 17:53:25 +0000 | [diff] [blame] | 1490 | // (~X) - (~Y) --> Y - X |
| 1491 | Value *X, *Y; |
| 1492 | if (match(Op0, m_Not(m_Value(X))) && match(Op1, m_Not(m_Value(Y)))) |
| 1493 | return BinaryOperator::CreateSub(Y, X); |
| 1494 | |
Sanjay Patel | 577c705 | 2018-07-29 18:13:16 +0000 | [diff] [blame] | 1495 | // (X + -1) - Y --> ~Y + X |
| 1496 | if (match(Op0, m_OneUse(m_Add(m_Value(X), m_AllOnes())))) |
| 1497 | return BinaryOperator::CreateAdd(Builder.CreateNot(Op1), X); |
| 1498 | |
| 1499 | // Y - (X + 1) --> ~X + Y |
| 1500 | if (match(Op1, m_OneUse(m_Add(m_Value(X), m_One())))) |
| 1501 | return BinaryOperator::CreateAdd(Builder.CreateNot(X), Op0); |
| 1502 | |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1503 | if (Constant *C = dyn_cast<Constant>(Op0)) { |
Sanjay Patel | 3bd957b | 2018-06-03 16:35:26 +0000 | [diff] [blame] | 1504 | bool IsNegate = match(C, m_ZeroInt()); |
Sanjay Patel | b6404a8 | 2017-12-06 21:22:57 +0000 | [diff] [blame] | 1505 | Value *X; |
Sanjay Patel | 3bd957b | 2018-06-03 16:35:26 +0000 | [diff] [blame] | 1506 | if (match(Op1, m_ZExt(m_Value(X))) && X->getType()->isIntOrIntVectorTy(1)) { |
| 1507 | // 0 - (zext bool) --> sext bool |
| 1508 | // C - (zext bool) --> bool ? C - 1 : C |
| 1509 | if (IsNegate) |
| 1510 | return CastInst::CreateSExtOrBitCast(X, I.getType()); |
Sanjay Patel | b6404a8 | 2017-12-06 21:22:57 +0000 | [diff] [blame] | 1511 | return SelectInst::Create(X, SubOne(C), C); |
Sanjay Patel | 3bd957b | 2018-06-03 16:35:26 +0000 | [diff] [blame] | 1512 | } |
| 1513 | if (match(Op1, m_SExt(m_Value(X))) && X->getType()->isIntOrIntVectorTy(1)) { |
| 1514 | // 0 - (sext bool) --> zext bool |
| 1515 | // C - (sext bool) --> bool ? C + 1 : C |
| 1516 | if (IsNegate) |
| 1517 | return CastInst::CreateZExtOrBitCast(X, I.getType()); |
| 1518 | return SelectInst::Create(X, AddOne(C), C); |
| 1519 | } |
Sanjay Patel | b6404a8 | 2017-12-06 21:22:57 +0000 | [diff] [blame] | 1520 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1521 | // C - ~X == X + (1+C) |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1522 | if (match(Op1, m_Not(m_Value(X)))) |
| 1523 | return BinaryOperator::CreateAdd(X, AddOne(C)); |
| 1524 | |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1525 | // Try to fold constant sub into select arguments. |
| 1526 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) |
| 1527 | if (Instruction *R = FoldOpIntoSelect(I, SI)) |
| 1528 | return R; |
| 1529 | |
Craig Topper | fb71b7d | 2017-04-14 19:20:12 +0000 | [diff] [blame] | 1530 | // Try to fold constant sub into PHI values. |
| 1531 | if (PHINode *PN = dyn_cast<PHINode>(Op1)) |
| 1532 | if (Instruction *R = foldOpIntoPhi(I, PN)) |
| 1533 | return R; |
| 1534 | |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1535 | // C-(X+C2) --> (C-C2)-X |
| 1536 | Constant *C2; |
| 1537 | if (match(Op1, m_Add(m_Value(X), m_Constant(C2)))) |
| 1538 | return BinaryOperator::CreateSub(ConstantExpr::getSub(C, C2), X); |
Benjamin Kramer | 72196f3 | 2014-01-19 15:24:22 +0000 | [diff] [blame] | 1539 | } |
| 1540 | |
Sanjay Patel | 6d6eca5 | 2016-10-14 16:31:54 +0000 | [diff] [blame] | 1541 | const APInt *Op0C; |
| 1542 | if (match(Op0, m_APInt(Op0C))) { |
| 1543 | unsigned BitWidth = I.getType()->getScalarSizeInBits(); |
| 1544 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1545 | // -(X >>u 31) -> (X >>s 31) |
| 1546 | // -(X >>s 31) -> (X >>u 31) |
Craig Topper | 73ba1c8 | 2017-06-07 07:40:37 +0000 | [diff] [blame] | 1547 | if (Op0C->isNullValue()) { |
David Majnemer | 72a643d | 2014-11-03 05:53:55 +0000 | [diff] [blame] | 1548 | Value *X; |
Sanjay Patel | 6d6eca5 | 2016-10-14 16:31:54 +0000 | [diff] [blame] | 1549 | const APInt *ShAmt; |
| 1550 | if (match(Op1, m_LShr(m_Value(X), m_APInt(ShAmt))) && |
| 1551 | *ShAmt == BitWidth - 1) { |
| 1552 | Value *ShAmtOp = cast<Instruction>(Op1)->getOperand(1); |
| 1553 | return BinaryOperator::CreateAShr(X, ShAmtOp); |
| 1554 | } |
| 1555 | if (match(Op1, m_AShr(m_Value(X), m_APInt(ShAmt))) && |
| 1556 | *ShAmt == BitWidth - 1) { |
| 1557 | Value *ShAmtOp = cast<Instruction>(Op1)->getOperand(1); |
| 1558 | return BinaryOperator::CreateLShr(X, ShAmtOp); |
| 1559 | } |
Craig Topper | 3b768e86 | 2018-05-23 17:29:03 +0000 | [diff] [blame] | 1560 | |
| 1561 | if (Op1->hasOneUse()) { |
| 1562 | Value *LHS, *RHS; |
| 1563 | SelectPatternFlavor SPF = matchSelectPattern(Op1, LHS, RHS).Flavor; |
| 1564 | if (SPF == SPF_ABS || SPF == SPF_NABS) { |
| 1565 | // This is a negate of an ABS/NABS pattern. Just swap the operands |
| 1566 | // of the select. |
| 1567 | SelectInst *SI = cast<SelectInst>(Op1); |
| 1568 | Value *TrueVal = SI->getTrueValue(); |
| 1569 | Value *FalseVal = SI->getFalseValue(); |
| 1570 | SI->setTrueValue(FalseVal); |
| 1571 | SI->setFalseValue(TrueVal); |
| 1572 | // Don't swap prof metadata, we didn't change the branch behavior. |
| 1573 | return replaceInstUsesWith(I, SI); |
| 1574 | } |
| 1575 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1576 | } |
Matthias Braun | ec68334 | 2015-04-30 22:04:26 +0000 | [diff] [blame] | 1577 | |
| 1578 | // Turn this into a xor if LHS is 2^n-1 and the remaining bits are known |
| 1579 | // zero. |
Craig Topper | b4da684 | 2017-04-06 21:06:03 +0000 | [diff] [blame] | 1580 | if (Op0C->isMask()) { |
Craig Topper | 8205a1a | 2017-05-24 16:53:07 +0000 | [diff] [blame] | 1581 | KnownBits RHSKnown = computeKnownBits(Op1, 0, &I); |
Craig Topper | b45eabc | 2017-04-26 16:39:58 +0000 | [diff] [blame] | 1582 | if ((*Op0C | RHSKnown.Zero).isAllOnesValue()) |
Sanjay Patel | 6d6eca5 | 2016-10-14 16:31:54 +0000 | [diff] [blame] | 1583 | return BinaryOperator::CreateXor(Op1, Op0); |
Matthias Braun | ec68334 | 2015-04-30 22:04:26 +0000 | [diff] [blame] | 1584 | } |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1585 | } |
| 1586 | |
David Majnemer | 72a643d | 2014-11-03 05:53:55 +0000 | [diff] [blame] | 1587 | { |
Suyog Sarda | cba4b1d | 2014-10-08 08:37:49 +0000 | [diff] [blame] | 1588 | Value *Y; |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1589 | // X-(X+Y) == -Y X-(Y+X) == -Y |
Craig Topper | 98851ad | 2017-04-10 16:59:40 +0000 | [diff] [blame] | 1590 | if (match(Op1, m_c_Add(m_Specific(Op0), m_Value(Y)))) |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1591 | return BinaryOperator::CreateNeg(Y); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1592 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1593 | // (X-Y)-X == -Y |
| 1594 | if (match(Op0, m_Sub(m_Specific(Op1), m_Value(Y)))) |
| 1595 | return BinaryOperator::CreateNeg(Y); |
| 1596 | } |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1597 | |
Hiroshi Yamauchi | 0445e31 | 2017-07-26 21:54:43 +0000 | [diff] [blame] | 1598 | // (sub (or A, B), (xor A, B)) --> (and A, B) |
David Majnemer | 312c3e5 | 2014-10-19 08:32:32 +0000 | [diff] [blame] | 1599 | { |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1600 | Value *A, *B; |
David Majnemer | 312c3e5 | 2014-10-19 08:32:32 +0000 | [diff] [blame] | 1601 | if (match(Op1, m_Xor(m_Value(A), m_Value(B))) && |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1602 | match(Op0, m_c_Or(m_Specific(A), m_Specific(B)))) |
David Majnemer | 312c3e5 | 2014-10-19 08:32:32 +0000 | [diff] [blame] | 1603 | return BinaryOperator::CreateAnd(A, B); |
| 1604 | } |
| 1605 | |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1606 | { |
| 1607 | Value *Y; |
David Majnemer | 72a643d | 2014-11-03 05:53:55 +0000 | [diff] [blame] | 1608 | // ((X | Y) - X) --> (~X & Y) |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1609 | if (match(Op0, m_OneUse(m_c_Or(m_Value(Y), m_Specific(Op1))))) |
David Majnemer | 72a643d | 2014-11-03 05:53:55 +0000 | [diff] [blame] | 1610 | return BinaryOperator::CreateAnd( |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1611 | Y, Builder.CreateNot(Op1, Op1->getName() + ".not")); |
David Majnemer | 72a643d | 2014-11-03 05:53:55 +0000 | [diff] [blame] | 1612 | } |
| 1613 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1614 | if (Op1->hasOneUse()) { |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1615 | Value *X = nullptr, *Y = nullptr, *Z = nullptr; |
| 1616 | Constant *C = nullptr; |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1617 | |
| 1618 | // (X - (Y - Z)) --> (X + (Z - Y)). |
| 1619 | if (match(Op1, m_Sub(m_Value(Y), m_Value(Z)))) |
| 1620 | return BinaryOperator::CreateAdd(Op0, |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1621 | Builder.CreateSub(Z, Y, Op1->getName())); |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1622 | |
| 1623 | // (X - (X & Y)) --> (X & ~Y) |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1624 | if (match(Op1, m_c_And(m_Value(Y), m_Specific(Op0)))) |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1625 | return BinaryOperator::CreateAnd(Op0, |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1626 | Builder.CreateNot(Y, Y->getName() + ".not")); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1627 | |
David Majnemer | bdeef60 | 2014-07-02 06:07:09 +0000 | [diff] [blame] | 1628 | // 0 - (X sdiv C) -> (X sdiv -C) provided the negation doesn't overflow. |
| 1629 | if (match(Op1, m_SDiv(m_Value(X), m_Constant(C))) && match(Op0, m_Zero()) && |
David Majnemer | 0e6c986 | 2014-08-22 16:41:23 +0000 | [diff] [blame] | 1630 | C->isNotMinSignedValue() && !C->isOneValue()) |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1631 | return BinaryOperator::CreateSDiv(X, ConstantExpr::getNeg(C)); |
| 1632 | |
| 1633 | // 0 - (X << Y) -> (-X << Y) when X is freely negatable. |
| 1634 | if (match(Op1, m_Shl(m_Value(X), m_Value(Y))) && match(Op0, m_Zero())) |
| 1635 | if (Value *XNeg = dyn_castNegVal(X)) |
| 1636 | return BinaryOperator::CreateShl(XNeg, Y); |
| 1637 | |
Sanjay Patel | c6c5965 | 2016-10-14 15:24:31 +0000 | [diff] [blame] | 1638 | // Subtracting -1/0 is the same as adding 1/0: |
| 1639 | // sub [nsw] Op0, sext(bool Y) -> add [nsw] Op0, zext(bool Y) |
| 1640 | // 'nuw' is dropped in favor of the canonical form. |
| 1641 | if (match(Op1, m_SExt(m_Value(Y))) && |
| 1642 | Y->getType()->getScalarSizeInBits() == 1) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1643 | Value *Zext = Builder.CreateZExt(Y, I.getType()); |
Sanjay Patel | c6c5965 | 2016-10-14 15:24:31 +0000 | [diff] [blame] | 1644 | BinaryOperator *Add = BinaryOperator::CreateAdd(Op0, Zext); |
| 1645 | Add->setHasNoSignedWrap(I.hasNoSignedWrap()); |
| 1646 | return Add; |
| 1647 | } |
| 1648 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1649 | // X - A*-B -> X + A*B |
| 1650 | // X - -A*B -> X + A*B |
| 1651 | Value *A, *B; |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1652 | Constant *CI; |
| 1653 | if (match(Op1, m_c_Mul(m_Value(A), m_Neg(m_Value(B))))) |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1654 | return BinaryOperator::CreateAdd(Op0, Builder.CreateMul(A, B)); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1655 | |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1656 | // X - A*CI -> X + A*-CI |
Craig Topper | 0d830ff | 2017-04-10 18:09:25 +0000 | [diff] [blame] | 1657 | // No need to handle commuted multiply because multiply handling will |
| 1658 | // ensure constant will be move to the right hand side. |
| 1659 | if (match(Op1, m_Mul(m_Value(A), m_Constant(CI)))) { |
Craig Topper | bb4069e | 2017-07-07 23:16:26 +0000 | [diff] [blame] | 1660 | Value *NewMul = Builder.CreateMul(A, ConstantExpr::getNeg(CI)); |
Chris Lattner | 7d0e43f | 2011-02-10 05:14:58 +0000 | [diff] [blame] | 1661 | return BinaryOperator::CreateAdd(Op0, NewMul); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1662 | } |
| 1663 | } |
| 1664 | |
David Green | 1e44c3b | 2018-10-02 09:48:34 +0000 | [diff] [blame] | 1665 | { |
| 1666 | // ~A - Min/Max(~A, O) -> Max/Min(A, ~O) - A |
| 1667 | // ~A - Min/Max(O, ~A) -> Max/Min(A, ~O) - A |
| 1668 | // Min/Max(~A, O) - ~A -> A - Max/Min(A, ~O) |
| 1669 | // Min/Max(O, ~A) - ~A -> A - Max/Min(A, ~O) |
| 1670 | // So long as O here is freely invertible, this will be neutral or a win. |
| 1671 | Value *LHS, *RHS, *A; |
| 1672 | Value *NotA = Op0, *MinMax = Op1; |
| 1673 | SelectPatternFlavor SPF = matchSelectPattern(MinMax, LHS, RHS).Flavor; |
| 1674 | if (!SelectPatternResult::isMinOrMax(SPF)) { |
| 1675 | NotA = Op1; |
| 1676 | MinMax = Op0; |
| 1677 | SPF = matchSelectPattern(MinMax, LHS, RHS).Flavor; |
| 1678 | } |
| 1679 | if (SelectPatternResult::isMinOrMax(SPF) && |
| 1680 | match(NotA, m_Not(m_Value(A))) && (NotA == LHS || NotA == RHS)) { |
| 1681 | if (NotA == LHS) |
| 1682 | std::swap(LHS, RHS); |
| 1683 | // LHS is now O above and expected to have at least 2 uses (the min/max) |
| 1684 | // NotA is epected to have 2 uses from the min/max and 1 from the sub. |
| 1685 | if (IsFreeToInvert(LHS, !LHS->hasNUsesOrMore(3)) && |
| 1686 | !NotA->hasNUsesOrMore(4)) { |
| 1687 | // Note: We don't generate the inverse max/min, just create the not of |
| 1688 | // it and let other folds do the rest. |
| 1689 | Value *Not = Builder.CreateNot(MinMax); |
| 1690 | if (NotA == Op0) |
| 1691 | return BinaryOperator::CreateSub(Not, A); |
| 1692 | else |
| 1693 | return BinaryOperator::CreateSub(A, Not); |
| 1694 | } |
| 1695 | } |
| 1696 | } |
| 1697 | |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1698 | // Optimize pointer differences into the same array into a size. Consider: |
| 1699 | // &A[10] - &A[0]: we should compile this to "10". |
Mehdi Amini | a28d91d | 2015-03-10 02:37:25 +0000 | [diff] [blame] | 1700 | Value *LHSOp, *RHSOp; |
| 1701 | if (match(Op0, m_PtrToInt(m_Value(LHSOp))) && |
| 1702 | match(Op1, m_PtrToInt(m_Value(RHSOp)))) |
| 1703 | if (Value *Res = OptimizePointerDifference(LHSOp, RHSOp, I.getType())) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1704 | return replaceInstUsesWith(I, Res); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1705 | |
Mehdi Amini | a28d91d | 2015-03-10 02:37:25 +0000 | [diff] [blame] | 1706 | // trunc(p)-trunc(q) -> trunc(p-q) |
| 1707 | if (match(Op0, m_Trunc(m_PtrToInt(m_Value(LHSOp)))) && |
| 1708 | match(Op1, m_Trunc(m_PtrToInt(m_Value(RHSOp))))) |
| 1709 | if (Value *Res = OptimizePointerDifference(LHSOp, RHSOp, I.getType())) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1710 | return replaceInstUsesWith(I, Res); |
Michael Ilseman | 9fc0f25 | 2012-12-12 20:57:53 +0000 | [diff] [blame] | 1711 | |
Sanjay Patel | 3cd1aa8 | 2018-06-06 21:58:12 +0000 | [diff] [blame] | 1712 | // Canonicalize a shifty way to code absolute value to the common pattern. |
| 1713 | // There are 2 potential commuted variants. |
| 1714 | // We're relying on the fact that we only do this transform when the shift has |
| 1715 | // exactly 2 uses and the xor has exactly 1 use (otherwise, we might increase |
| 1716 | // instructions). |
| 1717 | Value *A; |
| 1718 | const APInt *ShAmt; |
| 1719 | Type *Ty = I.getType(); |
| 1720 | if (match(Op1, m_AShr(m_Value(A), m_APInt(ShAmt))) && |
| 1721 | Op1->hasNUses(2) && *ShAmt == Ty->getScalarSizeInBits() - 1 && |
| 1722 | match(Op0, m_OneUse(m_c_Xor(m_Specific(A), m_Specific(Op1))))) { |
| 1723 | // B = ashr i32 A, 31 ; smear the sign bit |
| 1724 | // sub (xor A, B), B ; flip bits if negative and subtract -1 (add 1) |
| 1725 | // --> (A < 0) ? -A : A |
| 1726 | Value *Cmp = Builder.CreateICmpSLT(A, ConstantInt::getNullValue(Ty)); |
| 1727 | // Copy the nuw/nsw flags from the sub to the negate. |
| 1728 | Value *Neg = Builder.CreateNeg(A, "", I.hasNoUnsignedWrap(), |
| 1729 | I.hasNoSignedWrap()); |
| 1730 | return SelectInst::Create(Cmp, Neg, A); |
| 1731 | } |
| 1732 | |
Craig Topper | 2da7381 | 2018-09-15 18:54:10 +0000 | [diff] [blame] | 1733 | if (Instruction *Ext = narrowMathIfNoOverflow(I)) |
| 1734 | return Ext; |
| 1735 | |
David Majnemer | 57d5bc8 | 2014-08-19 23:36:30 +0000 | [diff] [blame] | 1736 | bool Changed = false; |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1737 | if (!I.hasNoSignedWrap() && willNotOverflowSignedSub(Op0, Op1, I)) { |
David Majnemer | 57d5bc8 | 2014-08-19 23:36:30 +0000 | [diff] [blame] | 1738 | Changed = true; |
| 1739 | I.setHasNoSignedWrap(true); |
| 1740 | } |
Craig Topper | 2b1fc32 | 2017-05-22 06:25:31 +0000 | [diff] [blame] | 1741 | if (!I.hasNoUnsignedWrap() && willNotOverflowUnsignedSub(Op0, Op1, I)) { |
David Majnemer | 42158f3 | 2014-08-20 07:17:31 +0000 | [diff] [blame] | 1742 | Changed = true; |
| 1743 | I.setHasNoUnsignedWrap(true); |
| 1744 | } |
David Majnemer | 57d5bc8 | 2014-08-19 23:36:30 +0000 | [diff] [blame] | 1745 | |
| 1746 | return Changed ? &I : nullptr; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1747 | } |
| 1748 | |
| 1749 | Instruction *InstCombiner::visitFSub(BinaryOperator &I) { |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1750 | if (Value *V = SimplifyFSubInst(I.getOperand(0), I.getOperand(1), |
| 1751 | I.getFastMathFlags(), |
Craig Topper | a420562 | 2017-06-09 03:21:29 +0000 | [diff] [blame] | 1752 | SQ.getWithInstruction(&I))) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1753 | return replaceInstUsesWith(I, V); |
Michael Ilseman | d5787be | 2012-12-12 00:28:32 +0000 | [diff] [blame] | 1754 | |
Sanjay Patel | 79dceb2 | 2018-10-03 15:20:58 +0000 | [diff] [blame^] | 1755 | if (Instruction *X = foldVectorBinop(I)) |
Sanjay Patel | bbc6d60 | 2018-06-02 16:27:44 +0000 | [diff] [blame] | 1756 | return X; |
| 1757 | |
Sanjay Patel | a9ca709 | 2018-04-06 17:24:08 +0000 | [diff] [blame] | 1758 | // Subtraction from -0.0 is the canonical form of fneg. |
| 1759 | // fsub nsz 0, X ==> fsub nsz -0.0, X |
Sanjay Patel | 7b0fc75 | 2018-06-21 17:06:36 +0000 | [diff] [blame] | 1760 | Value *Op0 = I.getOperand(0), *Op1 = I.getOperand(1); |
Sanjay Patel | a9ca709 | 2018-04-06 17:24:08 +0000 | [diff] [blame] | 1761 | if (I.hasNoSignedZeros() && match(Op0, m_PosZeroFP())) |
| 1762 | return BinaryOperator::CreateFNegFMF(Op1, &I); |
Sanjay Patel | 03e2526 | 2018-04-05 21:37:17 +0000 | [diff] [blame] | 1763 | |
Sanjay Patel | a194b2d | 2018-08-08 14:29:08 +0000 | [diff] [blame] | 1764 | Value *X, *Y; |
| 1765 | Constant *C; |
| 1766 | |
| 1767 | // Fold negation into constant operand. This is limited with one-use because |
| 1768 | // fneg is assumed better for analysis and cheaper in codegen than fmul/fdiv. |
| 1769 | // -(X * C) --> X * (-C) |
| 1770 | if (match(&I, m_FNeg(m_OneUse(m_FMul(m_Value(X), m_Constant(C)))))) |
| 1771 | return BinaryOperator::CreateFMulFMF(X, ConstantExpr::getFNeg(C), &I); |
| 1772 | // -(X / C) --> X / (-C) |
| 1773 | if (match(&I, m_FNeg(m_OneUse(m_FDiv(m_Value(X), m_Constant(C)))))) |
| 1774 | return BinaryOperator::CreateFDivFMF(X, ConstantExpr::getFNeg(C), &I); |
| 1775 | // -(C / X) --> (-C) / X |
| 1776 | if (match(&I, m_FNeg(m_OneUse(m_FDiv(m_Constant(C), m_Value(X)))))) |
| 1777 | return BinaryOperator::CreateFDivFMF(ConstantExpr::getFNeg(C), X, &I); |
| 1778 | |
Sanjay Patel | a9ca709 | 2018-04-06 17:24:08 +0000 | [diff] [blame] | 1779 | // If Op0 is not -0.0 or we can ignore -0.0: Z - (X - Y) --> Z + (Y - X) |
Sanjay Patel | 04683de | 2018-04-05 23:21:15 +0000 | [diff] [blame] | 1780 | // Canonicalize to fadd to make analysis easier. |
| 1781 | // This can also help codegen because fadd is commutative. |
Sanjay Patel | a9ca709 | 2018-04-06 17:24:08 +0000 | [diff] [blame] | 1782 | // Note that if this fsub was really an fneg, the fadd with -0.0 will get |
| 1783 | // killed later. We still limit that particular transform with 'hasOneUse' |
| 1784 | // because an fneg is assumed better/cheaper than a generic fsub. |
Sanjay Patel | 04683de | 2018-04-05 23:21:15 +0000 | [diff] [blame] | 1785 | if (I.hasNoSignedZeros() || CannotBeNegativeZero(Op0, SQ.TLI)) { |
| 1786 | if (match(Op1, m_OneUse(m_FSub(m_Value(X), m_Value(Y))))) { |
| 1787 | Value *NewSub = Builder.CreateFSubFMF(Y, X, &I); |
| 1788 | return BinaryOperator::CreateFAddFMF(Op0, NewSub, &I); |
| 1789 | } |
| 1790 | } |
| 1791 | |
Stephen Lin | a9b57f6 | 2013-07-20 07:13:13 +0000 | [diff] [blame] | 1792 | if (isa<Constant>(Op0)) |
| 1793 | if (SelectInst *SI = dyn_cast<SelectInst>(Op1)) |
| 1794 | if (Instruction *NV = FoldOpIntoSelect(I, SI)) |
| 1795 | return NV; |
| 1796 | |
Sanjay Patel | deaf4f3 | 2018-04-05 17:06:45 +0000 | [diff] [blame] | 1797 | // X - C --> X + (-C) |
Sanjay Patel | ceb595b | 2018-05-30 23:55:12 +0000 | [diff] [blame] | 1798 | // But don't transform constant expressions because there's an inverse fold |
| 1799 | // for X + (-Y) --> X - Y. |
Sanjay Patel | ceb595b | 2018-05-30 23:55:12 +0000 | [diff] [blame] | 1800 | if (match(Op1, m_Constant(C)) && !isa<ConstantExpr>(Op1)) |
Sanjay Patel | deaf4f3 | 2018-04-05 17:06:45 +0000 | [diff] [blame] | 1801 | return BinaryOperator::CreateFAddFMF(Op0, ConstantExpr::getFNeg(C), &I); |
Fangrui Song | f78650a | 2018-07-30 19:41:25 +0000 | [diff] [blame] | 1802 | |
Sanjay Patel | deaf4f3 | 2018-04-05 17:06:45 +0000 | [diff] [blame] | 1803 | // X - (-Y) --> X + Y |
Sanjay Patel | deaf4f3 | 2018-04-05 17:06:45 +0000 | [diff] [blame] | 1804 | if (match(Op1, m_FNeg(m_Value(Y)))) |
| 1805 | return BinaryOperator::CreateFAddFMF(Op0, Y, &I); |
Sanjay Patel | 4a9116e | 2018-02-23 17:07:29 +0000 | [diff] [blame] | 1806 | |
Sanjay Patel | ff98682 | 2018-04-11 15:57:18 +0000 | [diff] [blame] | 1807 | // Similar to above, but look through a cast of the negated value: |
| 1808 | // X - (fptrunc(-Y)) --> X + fptrunc(Y) |
Sanjay Patel | ebec420 | 2018-08-09 15:07:13 +0000 | [diff] [blame] | 1809 | Type *Ty = I.getType(); |
| 1810 | if (match(Op1, m_OneUse(m_FPTrunc(m_FNeg(m_Value(Y)))))) |
| 1811 | return BinaryOperator::CreateFAddFMF(Op0, Builder.CreateFPTrunc(Y, Ty), &I); |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1812 | |
Sanjay Patel | ebec420 | 2018-08-09 15:07:13 +0000 | [diff] [blame] | 1813 | // X - (fpext(-Y)) --> X + fpext(Y) |
| 1814 | if (match(Op1, m_OneUse(m_FPExt(m_FNeg(m_Value(Y)))))) |
| 1815 | return BinaryOperator::CreateFAddFMF(Op0, Builder.CreateFPExt(Y, Ty), &I); |
| 1816 | |
| 1817 | // Handle special cases for FSub with selects feeding the operation |
Quentin Colombet | aa103b3 | 2017-09-20 17:32:16 +0000 | [diff] [blame] | 1818 | if (Value *V = SimplifySelectsFeedingBinaryOp(I, Op0, Op1)) |
| 1819 | return replaceInstUsesWith(I, V); |
| 1820 | |
Warren Ristow | 8b2f27c | 2018-04-14 19:18:28 +0000 | [diff] [blame] | 1821 | if (I.hasAllowReassoc() && I.hasNoSignedZeros()) { |
Sanjay Patel | 2054dd7 | 2018-08-08 16:04:48 +0000 | [diff] [blame] | 1822 | // (Y - X) - Y --> -X |
| 1823 | if (match(Op0, m_FSub(m_Specific(Op1), m_Value(X)))) |
| 1824 | return BinaryOperator::CreateFNegFMF(X, &I); |
| 1825 | |
Sanjay Patel | fe83969 | 2018-08-08 16:19:22 +0000 | [diff] [blame] | 1826 | // Y - (X + Y) --> -X |
| 1827 | // Y - (Y + X) --> -X |
| 1828 | if (match(Op1, m_c_FAdd(m_Specific(Op0), m_Value(X)))) |
| 1829 | return BinaryOperator::CreateFNegFMF(X, &I); |
| 1830 | |
Sanjay Patel | 55accd7 | 2018-08-09 18:42:12 +0000 | [diff] [blame] | 1831 | // (X * C) - X --> X * (C - 1.0) |
| 1832 | if (match(Op0, m_FMul(m_Specific(Op1), m_Constant(C)))) { |
| 1833 | Constant *CSubOne = ConstantExpr::getFSub(C, ConstantFP::get(Ty, 1.0)); |
| 1834 | return BinaryOperator::CreateFMulFMF(Op1, CSubOne, &I); |
| 1835 | } |
| 1836 | // X - (X * C) --> X * (1.0 - C) |
| 1837 | if (match(Op1, m_FMul(m_Specific(Op0), m_Constant(C)))) { |
| 1838 | Constant *OneSubC = ConstantExpr::getFSub(ConstantFP::get(Ty, 1.0), C); |
| 1839 | return BinaryOperator::CreateFMulFMF(Op0, OneSubC, &I); |
| 1840 | } |
| 1841 | |
Sanjay Patel | dc185ee | 2018-08-12 15:48:26 +0000 | [diff] [blame] | 1842 | if (Instruction *F = factorizeFAddFSub(I, Builder)) |
| 1843 | return F; |
| 1844 | |
Sanjay Patel | 2054dd7 | 2018-08-08 16:04:48 +0000 | [diff] [blame] | 1845 | // TODO: This performs reassociative folds for FP ops. Some fraction of the |
| 1846 | // functionality has been subsumed by simple pattern matching here and in |
| 1847 | // InstSimplify. We should let a dedicated reassociation pass handle more |
| 1848 | // complex pattern matching and remove this from InstCombine. |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 1849 | if (Value *V = FAddCombine(Builder).simplify(&I)) |
Sanjay Patel | 4b19880 | 2016-02-01 22:23:39 +0000 | [diff] [blame] | 1850 | return replaceInstUsesWith(I, V); |
Shuxin Yang | 37a1efe | 2012-12-18 23:10:12 +0000 | [diff] [blame] | 1851 | } |
| 1852 | |
Craig Topper | f40110f | 2014-04-25 05:29:35 +0000 | [diff] [blame] | 1853 | return nullptr; |
Chris Lattner | 82aa888 | 2010-01-05 07:18:46 +0000 | [diff] [blame] | 1854 | } |