1 //===- InstCombineInternal.h - InstCombine pass internals -------*- C++ -*-===//
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 /// \file
11 ///
12 /// This file provides internal interfaces used to implement the InstCombine.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #ifndef LLVM_LIB_TRANSFORMS_INSTCOMBINE_INSTCOMBINEINTERNAL_H
17 #define LLVM_LIB_TRANSFORMS_INSTCOMBINE_INSTCOMBINEINTERNAL_H
18
19 #include "llvm/ADT/ArrayRef.h"
20 #include "llvm/Analysis/AliasAnalysis.h"
21 #include "llvm/Analysis/InstructionSimplify.h"
22 #include "llvm/Analysis/TargetFolder.h"
23 #include "llvm/Analysis/ValueTracking.h"
24 #include "llvm/IR/Argument.h"
25 #include "llvm/IR/BasicBlock.h"
26 #include "llvm/IR/Constant.h"
27 #include "llvm/IR/Constants.h"
28 #include "llvm/IR/DerivedTypes.h"
29 #include "llvm/IR/IRBuilder.h"
30 #include "llvm/IR/InstVisitor.h"
31 #include "llvm/IR/InstrTypes.h"
32 #include "llvm/IR/Instruction.h"
33 #include "llvm/IR/IntrinsicInst.h"
34 #include "llvm/IR/Intrinsics.h"
35 #include "llvm/IR/PatternMatch.h"
36 #include "llvm/IR/Use.h"
37 #include "llvm/IR/Value.h"
38 #include "llvm/Support/Casting.h"
39 #include "llvm/Support/Compiler.h"
40 #include "llvm/Support/Debug.h"
41 #include "llvm/Support/KnownBits.h"
42 #include "llvm/Support/raw_ostream.h"
43 #include "llvm/Transforms/InstCombine/InstCombineWorklist.h"
44 #include "llvm/Transforms/Utils/Local.h"
45 #include <cassert>
46 #include <cstdint>
47
48 #define DEBUG_TYPE "instcombine"
49
50 using namespace llvm::PatternMatch;
51
52 namespace llvm {
53
54 class APInt;
55 class AssumptionCache;
56 class CallSite;
57 class DataLayout;
58 class DominatorTree;
59 class GEPOperator;
60 class GlobalVariable;
61 class LoopInfo;
62 class OptimizationRemarkEmitter;
63 class TargetLibraryInfo;
64 class User;
65
66 /// Assign a complexity or rank value to LLVM Values. This is used to reduce
67 /// the amount of pattern matching needed for compares and commutative
68 /// instructions. For example, if we have:
69 /// icmp ugt X, Constant
70 /// or
71 /// xor (add X, Constant), cast Z
72 ///
73 /// We do not have to consider the commuted variants of these patterns because
74 /// canonicalization based on complexity guarantees the above ordering.
75 ///
76 /// This routine maps IR values to various complexity ranks:
77 /// 0 -> undef
78 /// 1 -> Constants
79 /// 2 -> Other non-instructions
80 /// 3 -> Arguments
81 /// 4 -> Cast and (f)neg/not instructions
82 /// 5 -> Other instructions
getComplexity(Value * V)83 static inline unsigned getComplexity(Value *V) {
84 if (isa<Instruction>(V)) {
85 if (isa<CastInst>(V) || match(V, m_Neg(m_Value())) ||
86 match(V, m_Not(m_Value())) || match(V, m_FNeg(m_Value())))
87 return 4;
88 return 5;
89 }
90 if (isa<Argument>(V))
91 return 3;
92 return isa<Constant>(V) ? (isa<UndefValue>(V) ? 0 : 1) : 2;
93 }
94
95 /// Predicate canonicalization reduces the number of patterns that need to be
96 /// matched by other transforms. For example, we may swap the operands of a
97 /// conditional branch or select to create a compare with a canonical (inverted)
98 /// predicate which is then more likely to be matched with other values.
isCanonicalPredicate(CmpInst::Predicate Pred)99 static inline bool isCanonicalPredicate(CmpInst::Predicate Pred) {
100 switch (Pred) {
101 case CmpInst::ICMP_NE:
102 case CmpInst::ICMP_ULE:
103 case CmpInst::ICMP_SLE:
104 case CmpInst::ICMP_UGE:
105 case CmpInst::ICMP_SGE:
106 // TODO: There are 16 FCMP predicates. Should others be (not) canonical?
107 case CmpInst::FCMP_ONE:
108 case CmpInst::FCMP_OLE:
109 case CmpInst::FCMP_OGE:
110 return false;
111 default:
112 return true;
113 }
114 }
115
116 /// Return the source operand of a potentially bitcasted value while optionally
117 /// checking if it has one use. If there is no bitcast or the one use check is
118 /// not met, return the input value itself.
119 static inline Value *peekThroughBitcast(Value *V, bool OneUseOnly = false) {
120 if (auto *BitCast = dyn_cast<BitCastInst>(V))
121 if (!OneUseOnly || BitCast->hasOneUse())
122 return BitCast->getOperand(0);
123
124 // V is not a bitcast or V has more than one use and OneUseOnly is true.
125 return V;
126 }
127
128 /// Add one to a Constant
AddOne(Constant * C)129 static inline Constant *AddOne(Constant *C) {
130 return ConstantExpr::getAdd(C, ConstantInt::get(C->getType(), 1));
131 }
132
133 /// Subtract one from a Constant
SubOne(Constant * C)134 static inline Constant *SubOne(Constant *C) {
135 return ConstantExpr::getSub(C, ConstantInt::get(C->getType(), 1));
136 }
137
138 /// Return true if the specified value is free to invert (apply ~ to).
139 /// This happens in cases where the ~ can be eliminated. If WillInvertAllUses
140 /// is true, work under the assumption that the caller intends to remove all
141 /// uses of V and only keep uses of ~V.
IsFreeToInvert(Value * V,bool WillInvertAllUses)142 static inline bool IsFreeToInvert(Value *V, bool WillInvertAllUses) {
143 // ~(~(X)) -> X.
144 if (match(V, m_Not(m_Value())))
145 return true;
146
147 // Constants can be considered to be not'ed values.
148 if (isa<ConstantInt>(V))
149 return true;
150
151 // A vector of constant integers can be inverted easily.
152 if (V->getType()->isVectorTy() && isa<Constant>(V)) {
153 unsigned NumElts = V->getType()->getVectorNumElements();
154 for (unsigned i = 0; i != NumElts; ++i) {
155 Constant *Elt = cast<Constant>(V)->getAggregateElement(i);
156 if (!Elt)
157 return false;
158
159 if (isa<UndefValue>(Elt))
160 continue;
161
162 if (!isa<ConstantInt>(Elt))
163 return false;
164 }
165 return true;
166 }
167
168 // Compares can be inverted if all of their uses are being modified to use the
169 // ~V.
170 if (isa<CmpInst>(V))
171 return WillInvertAllUses;
172
173 // If `V` is of the form `A + Constant` then `-1 - V` can be folded into `(-1
174 // - Constant) - A` if we are willing to invert all of the uses.
175 if (BinaryOperator *BO = dyn_cast<BinaryOperator>(V))
176 if (BO->getOpcode() == Instruction::Add ||
177 BO->getOpcode() == Instruction::Sub)
178 if (isa<Constant>(BO->getOperand(0)) || isa<Constant>(BO->getOperand(1)))
179 return WillInvertAllUses;
180
181 // Selects with invertible operands are freely invertible
182 if (match(V, m_Select(m_Value(), m_Not(m_Value()), m_Not(m_Value()))))
183 return WillInvertAllUses;
184
185 return false;
186 }
187
188 /// Specific patterns of overflow check idioms that we match.
189 enum OverflowCheckFlavor {
190 OCF_UNSIGNED_ADD,
191 OCF_SIGNED_ADD,
192 OCF_UNSIGNED_SUB,
193 OCF_SIGNED_SUB,
194 OCF_UNSIGNED_MUL,
195 OCF_SIGNED_MUL,
196
197 OCF_INVALID
198 };
199
200 /// Returns the OverflowCheckFlavor corresponding to a overflow_with_op
201 /// intrinsic.
202 static inline OverflowCheckFlavor
IntrinsicIDToOverflowCheckFlavor(unsigned ID)203 IntrinsicIDToOverflowCheckFlavor(unsigned ID) {
204 switch (ID) {
205 default:
206 return OCF_INVALID;
207 case Intrinsic::uadd_with_overflow:
208 return OCF_UNSIGNED_ADD;
209 case Intrinsic::sadd_with_overflow:
210 return OCF_SIGNED_ADD;
211 case Intrinsic::usub_with_overflow:
212 return OCF_UNSIGNED_SUB;
213 case Intrinsic::ssub_with_overflow:
214 return OCF_SIGNED_SUB;
215 case Intrinsic::umul_with_overflow:
216 return OCF_UNSIGNED_MUL;
217 case Intrinsic::smul_with_overflow:
218 return OCF_SIGNED_MUL;
219 }
220 }
221
222 /// Some binary operators require special handling to avoid poison and undefined
223 /// behavior. If a constant vector has undef elements, replace those undefs with
224 /// identity constants if possible because those are always safe to execute.
225 /// If no identity constant exists, replace undef with some other safe constant.
getSafeVectorConstantForBinop(BinaryOperator::BinaryOps Opcode,Constant * In,bool IsRHSConstant)226 static inline Constant *getSafeVectorConstantForBinop(
227 BinaryOperator::BinaryOps Opcode, Constant *In, bool IsRHSConstant) {
228 assert(In->getType()->isVectorTy() && "Not expecting scalars here");
229
230 Type *EltTy = In->getType()->getVectorElementType();
231 auto *SafeC = ConstantExpr::getBinOpIdentity(Opcode, EltTy, IsRHSConstant);
232 if (!SafeC) {
233 // TODO: Should this be available as a constant utility function? It is
234 // similar to getBinOpAbsorber().
235 if (IsRHSConstant) {
236 switch (Opcode) {
237 case Instruction::SRem: // X % 1 = 0
238 case Instruction::URem: // X %u 1 = 0
239 SafeC = ConstantInt::get(EltTy, 1);
240 break;
241 case Instruction::FRem: // X % 1.0 (doesn't simplify, but it is safe)
242 SafeC = ConstantFP::get(EltTy, 1.0);
243 break;
244 default:
245 llvm_unreachable("Only rem opcodes have no identity constant for RHS");
246 }
247 } else {
248 switch (Opcode) {
249 case Instruction::Shl: // 0 << X = 0
250 case Instruction::LShr: // 0 >>u X = 0
251 case Instruction::AShr: // 0 >> X = 0
252 case Instruction::SDiv: // 0 / X = 0
253 case Instruction::UDiv: // 0 /u X = 0
254 case Instruction::SRem: // 0 % X = 0
255 case Instruction::URem: // 0 %u X = 0
256 case Instruction::Sub: // 0 - X (doesn't simplify, but it is safe)
257 case Instruction::FSub: // 0.0 - X (doesn't simplify, but it is safe)
258 case Instruction::FDiv: // 0.0 / X (doesn't simplify, but it is safe)
259 case Instruction::FRem: // 0.0 % X = 0
260 SafeC = Constant::getNullValue(EltTy);
261 break;
262 default:
263 llvm_unreachable("Expected to find identity constant for opcode");
264 }
265 }
266 }
267 assert(SafeC && "Must have safe constant for binop");
268 unsigned NumElts = In->getType()->getVectorNumElements();
269 SmallVector<Constant *, 16> Out(NumElts);
270 for (unsigned i = 0; i != NumElts; ++i) {
271 Constant *C = In->getAggregateElement(i);
272 Out[i] = isa<UndefValue>(C) ? SafeC : C;
273 }
274 return ConstantVector::get(Out);
275 }
276
277 /// The core instruction combiner logic.
278 ///
279 /// This class provides both the logic to recursively visit instructions and
280 /// combine them.
281 class LLVM_LIBRARY_VISIBILITY InstCombiner
282 : public InstVisitor<InstCombiner, Instruction *> {
283 // FIXME: These members shouldn't be public.
284 public:
285 /// A worklist of the instructions that need to be simplified.
286 InstCombineWorklist &Worklist;
287
288 /// An IRBuilder that automatically inserts new instructions into the
289 /// worklist.
290 using BuilderTy = IRBuilder<TargetFolder, IRBuilderCallbackInserter>;
291 BuilderTy &Builder;
292
293 private:
294 // Mode in which we are running the combiner.
295 const bool MinimizeSize;
296
297 /// Enable combines that trigger rarely but are costly in compiletime.
298 const bool ExpensiveCombines;
299
300 AliasAnalysis *AA;
301
302 // Required analyses.
303 AssumptionCache &AC;
304 TargetLibraryInfo &TLI;
305 DominatorTree &DT;
306 const DataLayout &DL;
307 const SimplifyQuery SQ;
308 OptimizationRemarkEmitter &ORE;
309
310 // Optional analyses. When non-null, these can both be used to do better
311 // combining and will be updated to reflect any changes.
312 LoopInfo *LI;
313
314 bool MadeIRChange = false;
315
316 public:
InstCombiner(InstCombineWorklist & Worklist,BuilderTy & Builder,bool MinimizeSize,bool ExpensiveCombines,AliasAnalysis * AA,AssumptionCache & AC,TargetLibraryInfo & TLI,DominatorTree & DT,OptimizationRemarkEmitter & ORE,const DataLayout & DL,LoopInfo * LI)317 InstCombiner(InstCombineWorklist &Worklist, BuilderTy &Builder,
318 bool MinimizeSize, bool ExpensiveCombines, AliasAnalysis *AA,
319 AssumptionCache &AC, TargetLibraryInfo &TLI, DominatorTree &DT,
320 OptimizationRemarkEmitter &ORE, const DataLayout &DL,
321 LoopInfo *LI)
322 : Worklist(Worklist), Builder(Builder), MinimizeSize(MinimizeSize),
323 ExpensiveCombines(ExpensiveCombines), AA(AA), AC(AC), TLI(TLI), DT(DT),
324 DL(DL), SQ(DL, &TLI, &DT, &AC), ORE(ORE), LI(LI) {}
325
326 /// Run the combiner over the entire worklist until it is empty.
327 ///
328 /// \returns true if the IR is changed.
329 bool run();
330
getAssumptionCache()331 AssumptionCache &getAssumptionCache() const { return AC; }
332
getDataLayout()333 const DataLayout &getDataLayout() const { return DL; }
334
getDominatorTree()335 DominatorTree &getDominatorTree() const { return DT; }
336
getLoopInfo()337 LoopInfo *getLoopInfo() const { return LI; }
338
getTargetLibraryInfo()339 TargetLibraryInfo &getTargetLibraryInfo() const { return TLI; }
340
341 // Visitation implementation - Implement instruction combining for different
342 // instruction types. The semantics are as follows:
343 // Return Value:
344 // null - No change was made
345 // I - Change was made, I is still valid, I may be dead though
346 // otherwise - Change was made, replace I with returned instruction
347 //
348 Instruction *visitAdd(BinaryOperator &I);
349 Instruction *visitFAdd(BinaryOperator &I);
350 Value *OptimizePointerDifference(Value *LHS, Value *RHS, Type *Ty);
351 Instruction *visitSub(BinaryOperator &I);
352 Instruction *visitFSub(BinaryOperator &I);
353 Instruction *visitMul(BinaryOperator &I);
354 Instruction *visitFMul(BinaryOperator &I);
355 Instruction *visitURem(BinaryOperator &I);
356 Instruction *visitSRem(BinaryOperator &I);
357 Instruction *visitFRem(BinaryOperator &I);
358 bool simplifyDivRemOfSelectWithZeroOp(BinaryOperator &I);
359 Instruction *commonRemTransforms(BinaryOperator &I);
360 Instruction *commonIRemTransforms(BinaryOperator &I);
361 Instruction *commonDivTransforms(BinaryOperator &I);
362 Instruction *commonIDivTransforms(BinaryOperator &I);
363 Instruction *visitUDiv(BinaryOperator &I);
364 Instruction *visitSDiv(BinaryOperator &I);
365 Instruction *visitFDiv(BinaryOperator &I);
366 Value *simplifyRangeCheck(ICmpInst *Cmp0, ICmpInst *Cmp1, bool Inverted);
367 Instruction *visitAnd(BinaryOperator &I);
368 Instruction *visitOr(BinaryOperator &I);
369 Instruction *visitXor(BinaryOperator &I);
370 Instruction *visitShl(BinaryOperator &I);
371 Instruction *visitAShr(BinaryOperator &I);
372 Instruction *visitLShr(BinaryOperator &I);
373 Instruction *commonShiftTransforms(BinaryOperator &I);
374 Instruction *visitFCmpInst(FCmpInst &I);
375 Instruction *visitICmpInst(ICmpInst &I);
376 Instruction *FoldShiftByConstant(Value *Op0, Constant *Op1,
377 BinaryOperator &I);
378 Instruction *commonCastTransforms(CastInst &CI);
379 Instruction *commonPointerCastTransforms(CastInst &CI);
380 Instruction *visitTrunc(TruncInst &CI);
381 Instruction *visitZExt(ZExtInst &CI);
382 Instruction *visitSExt(SExtInst &CI);
383 Instruction *visitFPTrunc(FPTruncInst &CI);
384 Instruction *visitFPExt(CastInst &CI);
385 Instruction *visitFPToUI(FPToUIInst &FI);
386 Instruction *visitFPToSI(FPToSIInst &FI);
387 Instruction *visitUIToFP(CastInst &CI);
388 Instruction *visitSIToFP(CastInst &CI);
389 Instruction *visitPtrToInt(PtrToIntInst &CI);
390 Instruction *visitIntToPtr(IntToPtrInst &CI);
391 Instruction *visitBitCast(BitCastInst &CI);
392 Instruction *visitAddrSpaceCast(AddrSpaceCastInst &CI);
393 Instruction *FoldItoFPtoI(Instruction &FI);
394 Instruction *visitSelectInst(SelectInst &SI);
395 Instruction *visitCallInst(CallInst &CI);
396 Instruction *visitInvokeInst(InvokeInst &II);
397
398 Instruction *SliceUpIllegalIntegerPHI(PHINode &PN);
399 Instruction *visitPHINode(PHINode &PN);
400 Instruction *visitGetElementPtrInst(GetElementPtrInst &GEP);
401 Instruction *visitAllocaInst(AllocaInst &AI);
402 Instruction *visitAllocSite(Instruction &FI);
403 Instruction *visitFree(CallInst &FI);
404 Instruction *visitLoadInst(LoadInst &LI);
405 Instruction *visitStoreInst(StoreInst &SI);
406 Instruction *visitBranchInst(BranchInst &BI);
407 Instruction *visitFenceInst(FenceInst &FI);
408 Instruction *visitSwitchInst(SwitchInst &SI);
409 Instruction *visitReturnInst(ReturnInst &RI);
410 Instruction *visitInsertValueInst(InsertValueInst &IV);
411 Instruction *visitInsertElementInst(InsertElementInst &IE);
412 Instruction *visitExtractElementInst(ExtractElementInst &EI);
413 Instruction *visitShuffleVectorInst(ShuffleVectorInst &SVI);
414 Instruction *visitExtractValueInst(ExtractValueInst &EV);
415 Instruction *visitLandingPadInst(LandingPadInst &LI);
416 Instruction *visitVAStartInst(VAStartInst &I);
417 Instruction *visitVACopyInst(VACopyInst &I);
418
419 /// Specify what to return for unhandled instructions.
visitInstruction(Instruction & I)420 Instruction *visitInstruction(Instruction &I) { return nullptr; }
421
422 /// True when DB dominates all uses of DI except UI.
423 /// UI must be in the same block as DI.
424 /// The routine checks that the DI parent and DB are different.
425 bool dominatesAllUses(const Instruction *DI, const Instruction *UI,
426 const BasicBlock *DB) const;
427
428 /// Try to replace select with select operand SIOpd in SI-ICmp sequence.
429 bool replacedSelectWithOperand(SelectInst *SI, const ICmpInst *Icmp,
430 const unsigned SIOpd);
431
432 /// Try to replace instruction \p I with value \p V which are pointers
433 /// in different address space.
434 /// \return true if successful.
435 bool replacePointer(Instruction &I, Value *V);
436
437 private:
438 bool shouldChangeType(unsigned FromBitWidth, unsigned ToBitWidth) const;
439 bool shouldChangeType(Type *From, Type *To) const;
440 Value *dyn_castNegVal(Value *V) const;
441 Type *FindElementAtOffset(PointerType *PtrTy, int64_t Offset,
442 SmallVectorImpl<Value *> &NewIndices);
443
444 /// Classify whether a cast is worth optimizing.
445 ///
446 /// This is a helper to decide whether the simplification of
447 /// logic(cast(A), cast(B)) to cast(logic(A, B)) should be performed.
448 ///
449 /// \param CI The cast we are interested in.
450 ///
451 /// \return true if this cast actually results in any code being generated and
452 /// if it cannot already be eliminated by some other transformation.
453 bool shouldOptimizeCast(CastInst *CI);
454
455 /// Try to optimize a sequence of instructions checking if an operation
456 /// on LHS and RHS overflows.
457 ///
458 /// If this overflow check is done via one of the overflow check intrinsics,
459 /// then CtxI has to be the call instruction calling that intrinsic. If this
460 /// overflow check is done by arithmetic followed by a compare, then CtxI has
461 /// to be the arithmetic instruction.
462 ///
463 /// If a simplification is possible, stores the simplified result of the
464 /// operation in OperationResult and result of the overflow check in
465 /// OverflowResult, and return true. If no simplification is possible,
466 /// returns false.
467 bool OptimizeOverflowCheck(OverflowCheckFlavor OCF, Value *LHS, Value *RHS,
468 Instruction &CtxI, Value *&OperationResult,
469 Constant *&OverflowResult);
470
471 Instruction *visitCallSite(CallSite CS);
472 Instruction *tryOptimizeCall(CallInst *CI);
473 bool transformConstExprCastCall(CallSite CS);
474 Instruction *transformCallThroughTrampoline(CallSite CS,
475 IntrinsicInst *Tramp);
476
477 /// Transform (zext icmp) to bitwise / integer operations in order to
478 /// eliminate it.
479 ///
480 /// \param ICI The icmp of the (zext icmp) pair we are interested in.
481 /// \parem CI The zext of the (zext icmp) pair we are interested in.
482 /// \param DoTransform Pass false to just test whether the given (zext icmp)
483 /// would be transformed. Pass true to actually perform the transformation.
484 ///
485 /// \return null if the transformation cannot be performed. If the
486 /// transformation can be performed the new instruction that replaces the
487 /// (zext icmp) pair will be returned (if \p DoTransform is false the
488 /// unmodified \p ICI will be returned in this case).
489 Instruction *transformZExtICmp(ICmpInst *ICI, ZExtInst &CI,
490 bool DoTransform = true);
491
492 Instruction *transformSExtICmp(ICmpInst *ICI, Instruction &CI);
493
willNotOverflowSignedAdd(const Value * LHS,const Value * RHS,const Instruction & CxtI)494 bool willNotOverflowSignedAdd(const Value *LHS, const Value *RHS,
495 const Instruction &CxtI) const {
496 return computeOverflowForSignedAdd(LHS, RHS, &CxtI) ==
497 OverflowResult::NeverOverflows;
498 }
499
willNotOverflowUnsignedAdd(const Value * LHS,const Value * RHS,const Instruction & CxtI)500 bool willNotOverflowUnsignedAdd(const Value *LHS, const Value *RHS,
501 const Instruction &CxtI) const {
502 return computeOverflowForUnsignedAdd(LHS, RHS, &CxtI) ==
503 OverflowResult::NeverOverflows;
504 }
505
willNotOverflowAdd(const Value * LHS,const Value * RHS,const Instruction & CxtI,bool IsSigned)506 bool willNotOverflowAdd(const Value *LHS, const Value *RHS,
507 const Instruction &CxtI, bool IsSigned) const {
508 return IsSigned ? willNotOverflowSignedAdd(LHS, RHS, CxtI)
509 : willNotOverflowUnsignedAdd(LHS, RHS, CxtI);
510 }
511
willNotOverflowSignedSub(const Value * LHS,const Value * RHS,const Instruction & CxtI)512 bool willNotOverflowSignedSub(const Value *LHS, const Value *RHS,
513 const Instruction &CxtI) const {
514 return computeOverflowForSignedSub(LHS, RHS, &CxtI) ==
515 OverflowResult::NeverOverflows;
516 }
517
willNotOverflowUnsignedSub(const Value * LHS,const Value * RHS,const Instruction & CxtI)518 bool willNotOverflowUnsignedSub(const Value *LHS, const Value *RHS,
519 const Instruction &CxtI) const {
520 return computeOverflowForUnsignedSub(LHS, RHS, &CxtI) ==
521 OverflowResult::NeverOverflows;
522 }
523
willNotOverflowSub(const Value * LHS,const Value * RHS,const Instruction & CxtI,bool IsSigned)524 bool willNotOverflowSub(const Value *LHS, const Value *RHS,
525 const Instruction &CxtI, bool IsSigned) const {
526 return IsSigned ? willNotOverflowSignedSub(LHS, RHS, CxtI)
527 : willNotOverflowUnsignedSub(LHS, RHS, CxtI);
528 }
529
willNotOverflowSignedMul(const Value * LHS,const Value * RHS,const Instruction & CxtI)530 bool willNotOverflowSignedMul(const Value *LHS, const Value *RHS,
531 const Instruction &CxtI) const {
532 return computeOverflowForSignedMul(LHS, RHS, &CxtI) ==
533 OverflowResult::NeverOverflows;
534 }
535
willNotOverflowUnsignedMul(const Value * LHS,const Value * RHS,const Instruction & CxtI)536 bool willNotOverflowUnsignedMul(const Value *LHS, const Value *RHS,
537 const Instruction &CxtI) const {
538 return computeOverflowForUnsignedMul(LHS, RHS, &CxtI) ==
539 OverflowResult::NeverOverflows;
540 }
541
willNotOverflowMul(const Value * LHS,const Value * RHS,const Instruction & CxtI,bool IsSigned)542 bool willNotOverflowMul(const Value *LHS, const Value *RHS,
543 const Instruction &CxtI, bool IsSigned) const {
544 return IsSigned ? willNotOverflowSignedMul(LHS, RHS, CxtI)
545 : willNotOverflowUnsignedMul(LHS, RHS, CxtI);
546 }
547
willNotOverflow(BinaryOperator::BinaryOps Opcode,const Value * LHS,const Value * RHS,const Instruction & CxtI,bool IsSigned)548 bool willNotOverflow(BinaryOperator::BinaryOps Opcode, const Value *LHS,
549 const Value *RHS, const Instruction &CxtI,
550 bool IsSigned) const {
551 switch (Opcode) {
552 case Instruction::Add: return willNotOverflowAdd(LHS, RHS, CxtI, IsSigned);
553 case Instruction::Sub: return willNotOverflowSub(LHS, RHS, CxtI, IsSigned);
554 case Instruction::Mul: return willNotOverflowMul(LHS, RHS, CxtI, IsSigned);
555 default: llvm_unreachable("Unexpected opcode for overflow query");
556 }
557 }
558
559 Value *EmitGEPOffset(User *GEP);
560 Instruction *scalarizePHI(ExtractElementInst &EI, PHINode *PN);
561 Instruction *foldCastedBitwiseLogic(BinaryOperator &I);
562 Instruction *narrowBinOp(TruncInst &Trunc);
563 Instruction *narrowMaskedBinOp(BinaryOperator &And);
564 Instruction *narrowMathIfNoOverflow(BinaryOperator &I);
565 Instruction *narrowRotate(TruncInst &Trunc);
566 Instruction *optimizeBitCastFromPhi(CastInst &CI, PHINode *PN);
567
568 /// Determine if a pair of casts can be replaced by a single cast.
569 ///
570 /// \param CI1 The first of a pair of casts.
571 /// \param CI2 The second of a pair of casts.
572 ///
573 /// \return 0 if the cast pair cannot be eliminated, otherwise returns an
574 /// Instruction::CastOps value for a cast that can replace the pair, casting
575 /// CI1->getSrcTy() to CI2->getDstTy().
576 ///
577 /// \see CastInst::isEliminableCastPair
578 Instruction::CastOps isEliminableCastPair(const CastInst *CI1,
579 const CastInst *CI2);
580
581 Value *foldAndOfICmps(ICmpInst *LHS, ICmpInst *RHS, Instruction &CxtI);
582 Value *foldOrOfICmps(ICmpInst *LHS, ICmpInst *RHS, Instruction &CxtI);
583 Value *foldXorOfICmps(ICmpInst *LHS, ICmpInst *RHS);
584
585 /// Optimize (fcmp)&(fcmp) or (fcmp)|(fcmp).
586 /// NOTE: Unlike most of instcombine, this returns a Value which should
587 /// already be inserted into the function.
588 Value *foldLogicOfFCmps(FCmpInst *LHS, FCmpInst *RHS, bool IsAnd);
589
590 Value *foldAndOrOfICmpsOfAndWithPow2(ICmpInst *LHS, ICmpInst *RHS,
591 bool JoinedByAnd, Instruction &CxtI);
592 Value *matchSelectFromAndOr(Value *A, Value *B, Value *C, Value *D);
593 Value *getSelectCondition(Value *A, Value *B);
594
595 public:
596 /// Inserts an instruction \p New before instruction \p Old
597 ///
598 /// Also adds the new instruction to the worklist and returns \p New so that
599 /// it is suitable for use as the return from the visitation patterns.
InsertNewInstBefore(Instruction * New,Instruction & Old)600 Instruction *InsertNewInstBefore(Instruction *New, Instruction &Old) {
601 assert(New && !New->getParent() &&
602 "New instruction already inserted into a basic block!");
603 BasicBlock *BB = Old.getParent();
604 BB->getInstList().insert(Old.getIterator(), New); // Insert inst
605 Worklist.Add(New);
606 return New;
607 }
608
609 /// Same as InsertNewInstBefore, but also sets the debug loc.
InsertNewInstWith(Instruction * New,Instruction & Old)610 Instruction *InsertNewInstWith(Instruction *New, Instruction &Old) {
611 New->setDebugLoc(Old.getDebugLoc());
612 return InsertNewInstBefore(New, Old);
613 }
614
615 /// A combiner-aware RAUW-like routine.
616 ///
617 /// This method is to be used when an instruction is found to be dead,
618 /// replaceable with another preexisting expression. Here we add all uses of
619 /// I to the worklist, replace all uses of I with the new value, then return
620 /// I, so that the inst combiner will know that I was modified.
replaceInstUsesWith(Instruction & I,Value * V)621 Instruction *replaceInstUsesWith(Instruction &I, Value *V) {
622 // If there are no uses to replace, then we return nullptr to indicate that
623 // no changes were made to the program.
624 if (I.use_empty()) return nullptr;
625
626 Worklist.AddUsersToWorkList(I); // Add all modified instrs to worklist.
627
628 // If we are replacing the instruction with itself, this must be in a
629 // segment of unreachable code, so just clobber the instruction.
630 if (&I == V)
631 V = UndefValue::get(I.getType());
632
633 LLVM_DEBUG(dbgs() << "IC: Replacing " << I << "\n"
634 << " with " << *V << '\n');
635
636 I.replaceAllUsesWith(V);
637 return &I;
638 }
639
640 /// Creates a result tuple for an overflow intrinsic \p II with a given
641 /// \p Result and a constant \p Overflow value.
CreateOverflowTuple(IntrinsicInst * II,Value * Result,Constant * Overflow)642 Instruction *CreateOverflowTuple(IntrinsicInst *II, Value *Result,
643 Constant *Overflow) {
644 Constant *V[] = {UndefValue::get(Result->getType()), Overflow};
645 StructType *ST = cast<StructType>(II->getType());
646 Constant *Struct = ConstantStruct::get(ST, V);
647 return InsertValueInst::Create(Struct, Result, 0);
648 }
649
650 /// Combiner aware instruction erasure.
651 ///
652 /// When dealing with an instruction that has side effects or produces a void
653 /// value, we can't rely on DCE to delete the instruction. Instead, visit
654 /// methods should return the value returned by this function.
eraseInstFromFunction(Instruction & I)655 Instruction *eraseInstFromFunction(Instruction &I) {
656 LLVM_DEBUG(dbgs() << "IC: ERASE " << I << '\n');
657 assert(I.use_empty() && "Cannot erase instruction that is used!");
658 salvageDebugInfo(I);
659
660 // Make sure that we reprocess all operands now that we reduced their
661 // use counts.
662 if (I.getNumOperands() < 8) {
663 for (Use &Operand : I.operands())
664 if (auto *Inst = dyn_cast<Instruction>(Operand))
665 Worklist.Add(Inst);
666 }
667 Worklist.Remove(&I);
668 I.eraseFromParent();
669 MadeIRChange = true;
670 return nullptr; // Don't do anything with FI
671 }
672
computeKnownBits(const Value * V,KnownBits & Known,unsigned Depth,const Instruction * CxtI)673 void computeKnownBits(const Value *V, KnownBits &Known,
674 unsigned Depth, const Instruction *CxtI) const {
675 llvm::computeKnownBits(V, Known, DL, Depth, &AC, CxtI, &DT);
676 }
677
computeKnownBits(const Value * V,unsigned Depth,const Instruction * CxtI)678 KnownBits computeKnownBits(const Value *V, unsigned Depth,
679 const Instruction *CxtI) const {
680 return llvm::computeKnownBits(V, DL, Depth, &AC, CxtI, &DT);
681 }
682
683 bool isKnownToBeAPowerOfTwo(const Value *V, bool OrZero = false,
684 unsigned Depth = 0,
685 const Instruction *CxtI = nullptr) {
686 return llvm::isKnownToBeAPowerOfTwo(V, DL, OrZero, Depth, &AC, CxtI, &DT);
687 }
688
689 bool MaskedValueIsZero(const Value *V, const APInt &Mask, unsigned Depth = 0,
690 const Instruction *CxtI = nullptr) const {
691 return llvm::MaskedValueIsZero(V, Mask, DL, Depth, &AC, CxtI, &DT);
692 }
693
694 unsigned ComputeNumSignBits(const Value *Op, unsigned Depth = 0,
695 const Instruction *CxtI = nullptr) const {
696 return llvm::ComputeNumSignBits(Op, DL, Depth, &AC, CxtI, &DT);
697 }
698
computeOverflowForUnsignedMul(const Value * LHS,const Value * RHS,const Instruction * CxtI)699 OverflowResult computeOverflowForUnsignedMul(const Value *LHS,
700 const Value *RHS,
701 const Instruction *CxtI) const {
702 return llvm::computeOverflowForUnsignedMul(LHS, RHS, DL, &AC, CxtI, &DT);
703 }
704
computeOverflowForSignedMul(const Value * LHS,const Value * RHS,const Instruction * CxtI)705 OverflowResult computeOverflowForSignedMul(const Value *LHS,
706 const Value *RHS,
707 const Instruction *CxtI) const {
708 return llvm::computeOverflowForSignedMul(LHS, RHS, DL, &AC, CxtI, &DT);
709 }
710
computeOverflowForUnsignedAdd(const Value * LHS,const Value * RHS,const Instruction * CxtI)711 OverflowResult computeOverflowForUnsignedAdd(const Value *LHS,
712 const Value *RHS,
713 const Instruction *CxtI) const {
714 return llvm::computeOverflowForUnsignedAdd(LHS, RHS, DL, &AC, CxtI, &DT);
715 }
716
computeOverflowForSignedAdd(const Value * LHS,const Value * RHS,const Instruction * CxtI)717 OverflowResult computeOverflowForSignedAdd(const Value *LHS,
718 const Value *RHS,
719 const Instruction *CxtI) const {
720 return llvm::computeOverflowForSignedAdd(LHS, RHS, DL, &AC, CxtI, &DT);
721 }
722
computeOverflowForUnsignedSub(const Value * LHS,const Value * RHS,const Instruction * CxtI)723 OverflowResult computeOverflowForUnsignedSub(const Value *LHS,
724 const Value *RHS,
725 const Instruction *CxtI) const {
726 return llvm::computeOverflowForUnsignedSub(LHS, RHS, DL, &AC, CxtI, &DT);
727 }
728
computeOverflowForSignedSub(const Value * LHS,const Value * RHS,const Instruction * CxtI)729 OverflowResult computeOverflowForSignedSub(const Value *LHS, const Value *RHS,
730 const Instruction *CxtI) const {
731 return llvm::computeOverflowForSignedSub(LHS, RHS, DL, &AC, CxtI, &DT);
732 }
733
734 /// Maximum size of array considered when transforming.
735 uint64_t MaxArraySizeForCombine;
736
737 private:
738 /// Performs a few simplifications for operators which are associative
739 /// or commutative.
740 bool SimplifyAssociativeOrCommutative(BinaryOperator &I);
741
742 /// Tries to simplify binary operations which some other binary
743 /// operation distributes over.
744 ///
745 /// It does this by either by factorizing out common terms (eg "(A*B)+(A*C)"
746 /// -> "A*(B+C)") or expanding out if this results in simplifications (eg: "A
747 /// & (B | C) -> (A&B) | (A&C)" if this is a win). Returns the simplified
748 /// value, or null if it didn't simplify.
749 Value *SimplifyUsingDistributiveLaws(BinaryOperator &I);
750
751 /// Tries to simplify add operations using the definition of remainder.
752 ///
753 /// The definition of remainder is X % C = X - (X / C ) * C. The add
754 /// expression X % C0 + (( X / C0 ) % C1) * C0 can be simplified to
755 /// X % (C0 * C1)
756 Value *SimplifyAddWithRemainder(BinaryOperator &I);
757
758 // Binary Op helper for select operations where the expression can be
759 // efficiently reorganized.
760 Value *SimplifySelectsFeedingBinaryOp(BinaryOperator &I, Value *LHS,
761 Value *RHS);
762
763 /// This tries to simplify binary operations by factorizing out common terms
764 /// (e. g. "(A*B)+(A*C)" -> "A*(B+C)").
765 Value *tryFactorization(BinaryOperator &, Instruction::BinaryOps, Value *,
766 Value *, Value *, Value *);
767
768 /// Match a select chain which produces one of three values based on whether
769 /// the LHS is less than, equal to, or greater than RHS respectively.
770 /// Return true if we matched a three way compare idiom. The LHS, RHS, Less,
771 /// Equal and Greater values are saved in the matching process and returned to
772 /// the caller.
773 bool matchThreeWayIntCompare(SelectInst *SI, Value *&LHS, Value *&RHS,
774 ConstantInt *&Less, ConstantInt *&Equal,
775 ConstantInt *&Greater);
776
777 /// Attempts to replace V with a simpler value based on the demanded
778 /// bits.
779 Value *SimplifyDemandedUseBits(Value *V, APInt DemandedMask, KnownBits &Known,
780 unsigned Depth, Instruction *CxtI);
781 bool SimplifyDemandedBits(Instruction *I, unsigned Op,
782 const APInt &DemandedMask, KnownBits &Known,
783 unsigned Depth = 0);
784
785 /// Helper routine of SimplifyDemandedUseBits. It computes KnownZero/KnownOne
786 /// bits. It also tries to handle simplifications that can be done based on
787 /// DemandedMask, but without modifying the Instruction.
788 Value *SimplifyMultipleUseDemandedBits(Instruction *I,
789 const APInt &DemandedMask,
790 KnownBits &Known,
791 unsigned Depth, Instruction *CxtI);
792
793 /// Helper routine of SimplifyDemandedUseBits. It tries to simplify demanded
794 /// bit for "r1 = shr x, c1; r2 = shl r1, c2" instruction sequence.
795 Value *simplifyShrShlDemandedBits(
796 Instruction *Shr, const APInt &ShrOp1, Instruction *Shl,
797 const APInt &ShlOp1, const APInt &DemandedMask, KnownBits &Known);
798
799 /// Tries to simplify operands to an integer instruction based on its
800 /// demanded bits.
801 bool SimplifyDemandedInstructionBits(Instruction &Inst);
802
803 Value *simplifyAMDGCNMemoryIntrinsicDemanded(IntrinsicInst *II,
804 APInt DemandedElts,
805 int DmaskIdx = -1,
806 int TFCIdx = -1);
807
808 Value *SimplifyDemandedVectorElts(Value *V, APInt DemandedElts,
809 APInt &UndefElts, unsigned Depth = 0);
810
811 /// Canonicalize the position of binops relative to shufflevector.
812 Instruction *foldVectorBinop(BinaryOperator &Inst);
813
814 /// Given a binary operator, cast instruction, or select which has a PHI node
815 /// as operand #0, see if we can fold the instruction into the PHI (which is
816 /// only possible if all operands to the PHI are constants).
817 Instruction *foldOpIntoPhi(Instruction &I, PHINode *PN);
818
819 /// Given an instruction with a select as one operand and a constant as the
820 /// other operand, try to fold the binary operator into the select arguments.
821 /// This also works for Cast instructions, which obviously do not have a
822 /// second operand.
823 Instruction *FoldOpIntoSelect(Instruction &Op, SelectInst *SI);
824
825 /// This is a convenience wrapper function for the above two functions.
826 Instruction *foldBinOpIntoSelectOrPhi(BinaryOperator &I);
827
828 Instruction *foldAddWithConstant(BinaryOperator &Add);
829
830 /// Try to rotate an operation below a PHI node, using PHI nodes for
831 /// its operands.
832 Instruction *FoldPHIArgOpIntoPHI(PHINode &PN);
833 Instruction *FoldPHIArgBinOpIntoPHI(PHINode &PN);
834 Instruction *FoldPHIArgGEPIntoPHI(PHINode &PN);
835 Instruction *FoldPHIArgLoadIntoPHI(PHINode &PN);
836 Instruction *FoldPHIArgZextsIntoPHI(PHINode &PN);
837
838 /// If an integer typed PHI has only one use which is an IntToPtr operation,
839 /// replace the PHI with an existing pointer typed PHI if it exists. Otherwise
840 /// insert a new pointer typed PHI and replace the original one.
841 Instruction *FoldIntegerTypedPHI(PHINode &PN);
842
843 /// Helper function for FoldPHIArgXIntoPHI() to set debug location for the
844 /// folded operation.
845 void PHIArgMergedDebugLoc(Instruction *Inst, PHINode &PN);
846
847 Instruction *foldGEPICmp(GEPOperator *GEPLHS, Value *RHS,
848 ICmpInst::Predicate Cond, Instruction &I);
849 Instruction *foldAllocaCmp(ICmpInst &ICI, const AllocaInst *Alloca,
850 const Value *Other);
851 Instruction *foldCmpLoadFromIndexedGlobal(GetElementPtrInst *GEP,
852 GlobalVariable *GV, CmpInst &ICI,
853 ConstantInt *AndCst = nullptr);
854 Instruction *foldFCmpIntToFPConst(FCmpInst &I, Instruction *LHSI,
855 Constant *RHSC);
856 Instruction *foldICmpAddOpConst(Value *X, const APInt &C,
857 ICmpInst::Predicate Pred);
858 Instruction *foldICmpWithCastAndCast(ICmpInst &ICI);
859
860 Instruction *foldICmpUsingKnownBits(ICmpInst &Cmp);
861 Instruction *foldICmpWithDominatingICmp(ICmpInst &Cmp);
862 Instruction *foldICmpWithConstant(ICmpInst &Cmp);
863 Instruction *foldICmpInstWithConstant(ICmpInst &Cmp);
864 Instruction *foldICmpInstWithConstantNotInt(ICmpInst &Cmp);
865 Instruction *foldICmpBinOp(ICmpInst &Cmp);
866 Instruction *foldICmpEquality(ICmpInst &Cmp);
867 Instruction *foldICmpWithZero(ICmpInst &Cmp);
868
869 Instruction *foldICmpSelectConstant(ICmpInst &Cmp, SelectInst *Select,
870 ConstantInt *C);
871 Instruction *foldICmpBitCastConstant(ICmpInst &Cmp, BitCastInst *Bitcast,
872 const APInt &C);
873 Instruction *foldICmpTruncConstant(ICmpInst &Cmp, TruncInst *Trunc,
874 const APInt &C);
875 Instruction *foldICmpAndConstant(ICmpInst &Cmp, BinaryOperator *And,
876 const APInt &C);
877 Instruction *foldICmpXorConstant(ICmpInst &Cmp, BinaryOperator *Xor,
878 const APInt &C);
879 Instruction *foldICmpOrConstant(ICmpInst &Cmp, BinaryOperator *Or,
880 const APInt &C);
881 Instruction *foldICmpMulConstant(ICmpInst &Cmp, BinaryOperator *Mul,
882 const APInt &C);
883 Instruction *foldICmpShlConstant(ICmpInst &Cmp, BinaryOperator *Shl,
884 const APInt &C);
885 Instruction *foldICmpShrConstant(ICmpInst &Cmp, BinaryOperator *Shr,
886 const APInt &C);
887 Instruction *foldICmpUDivConstant(ICmpInst &Cmp, BinaryOperator *UDiv,
888 const APInt &C);
889 Instruction *foldICmpDivConstant(ICmpInst &Cmp, BinaryOperator *Div,
890 const APInt &C);
891 Instruction *foldICmpSubConstant(ICmpInst &Cmp, BinaryOperator *Sub,
892 const APInt &C);
893 Instruction *foldICmpAddConstant(ICmpInst &Cmp, BinaryOperator *Add,
894 const APInt &C);
895 Instruction *foldICmpAndConstConst(ICmpInst &Cmp, BinaryOperator *And,
896 const APInt &C1);
897 Instruction *foldICmpAndShift(ICmpInst &Cmp, BinaryOperator *And,
898 const APInt &C1, const APInt &C2);
899 Instruction *foldICmpShrConstConst(ICmpInst &I, Value *ShAmt, const APInt &C1,
900 const APInt &C2);
901 Instruction *foldICmpShlConstConst(ICmpInst &I, Value *ShAmt, const APInt &C1,
902 const APInt &C2);
903
904 Instruction *foldICmpBinOpEqualityWithConstant(ICmpInst &Cmp,
905 BinaryOperator *BO,
906 const APInt &C);
907 Instruction *foldICmpIntrinsicWithConstant(ICmpInst &ICI, const APInt &C);
908
909 // Helpers of visitSelectInst().
910 Instruction *foldSelectExtConst(SelectInst &Sel);
911 Instruction *foldSelectOpOp(SelectInst &SI, Instruction *TI, Instruction *FI);
912 Instruction *foldSelectIntoOp(SelectInst &SI, Value *, Value *);
913 Instruction *foldSPFofSPF(Instruction *Inner, SelectPatternFlavor SPF1,
914 Value *A, Value *B, Instruction &Outer,
915 SelectPatternFlavor SPF2, Value *C);
916 Instruction *foldSelectInstWithICmp(SelectInst &SI, ICmpInst *ICI);
917
918 Instruction *OptAndOp(BinaryOperator *Op, ConstantInt *OpRHS,
919 ConstantInt *AndRHS, BinaryOperator &TheAnd);
920
921 Value *insertRangeTest(Value *V, const APInt &Lo, const APInt &Hi,
922 bool isSigned, bool Inside);
923 Instruction *PromoteCastOfAllocation(BitCastInst &CI, AllocaInst &AI);
924 bool mergeStoreIntoSuccessor(StoreInst &SI);
925
926 /// Given an 'or' instruction, check to see if it is part of a bswap idiom.
927 /// If so, return the equivalent bswap intrinsic.
928 Instruction *matchBSwap(BinaryOperator &Or);
929
930 Instruction *SimplifyAnyMemTransfer(AnyMemTransferInst *MI);
931 Instruction *SimplifyAnyMemSet(AnyMemSetInst *MI);
932
933 Value *EvaluateInDifferentType(Value *V, Type *Ty, bool isSigned);
934
935 /// Returns a value X such that Val = X * Scale, or null if none.
936 ///
937 /// If the multiplication is known not to overflow then NoSignedWrap is set.
938 Value *Descale(Value *Val, APInt Scale, bool &NoSignedWrap);
939 };
940
941 } // end namespace llvm
942
943 #undef DEBUG_TYPE
944
945 #endif // LLVM_LIB_TRANSFORMS_INSTCOMBINE_INSTCOMBINEINTERNAL_H
946